app.py

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)