|
|
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) |
|
|
|
