from transformers import pipeline, set_seed from flask import Flask, jsonify, request, render_template
app = Flask(name) generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
@app.route('/') def home(): return render_template('index.html')
@app.route('/get_answer', methods=['POST']) def get_answer(): set_seed(42) question = request.form['question'] answer = generator(question, max_length=50, do_sample=True, temperature=0.7)[0]['generated_text'] return jsonify({'answer': answer})
if name == 'main': app.run(debug=True)