| from flask import Flask, request, jsonify, render_template |
| from transformers import pipeline |
|
|
| app = Flask(__name__) |
|
|
| |
| generator = pipeline("text-generation", model="gpt2") |
|
|
| @app.route("/") |
| def index(): |
| return render_template("index.html") |
|
|
| @app.route("/generate-response", methods=["POST"]) |
| def generate_response(): |
| data = request.get_json() |
| user_input = data.get("text", "") |
|
|
| |
| result = generator(user_input, max_length=100, num_return_sequences=1) |
| response = result[0]["generated_text"] |
|
|
| return jsonify({"response": response}) |
|
|
| if __name__ == "__main__": |
| app.run(host="0.0.0.0", port=7860) |
|
|
