Spaces:
Sleeping
Sleeping
File size: 745 Bytes
abb28f8 303b2fe e79e34b abb28f8 ad9dfac ca835bb 303b2fe ff1f742 82ddab9 abb28f8 69f9211 303b2fe 0f5010f abb28f8 ad9dfac ff1f742 ad9dfac e79e34b ff1f742 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from flask import Flask, request, jsonify
from transformers import pipeline
import os
app = Flask(__name__)
pipe = pipeline("text2text-generation", model="memorease/flan5_memorease", token=os.environ.get("HF_TOKEN"))
@app.route("/ask", methods=["POST"])
def ask_question():
try:
input_text = request.json.get("text")
result = pipe(f"Generate a question about: {input_text}")[0]["generated_text"]
return jsonify({"question": result})
except Exception as e:
return jsonify({"error": str(e)}), 500
@app.route("/", methods=["GET"])
def root_check():
return jsonify({"status": "running"})
if __name__ == "__main__":
port = int(os.environ.get("PORT", 7860))
app.run(host="0.0.0.0", port=port)
|