Update app.py

app.py

@@ -1,18 +1,18 @@
-from flask import Flask, request, jsonify
 from transformers import pipeline
 
+summarizer = pipeline("summarization", model="t5-base", framework="pt")  # Force PyTorch
+
+from flask import Flask, request, jsonify
+
 app = Flask(__name__)
-summarizer = pipeline("summarization", model="t5-base")
 
 @app.route("/summarize", methods=["POST"])
 def summarize():
     data = request.json
     text = data.get("text", "")
-    if not text:
-        return jsonify({"error": "No text provided"}), 400
-
-    summary = summarizer(text, max_length=200, min_length=50, do_sample=False)
-    return jsonify({"summary": summary[0]['summary_text']})
+    summary = summarizer(text, max_length=150, min_length=30, do_sample=False)
+    return jsonify({"summary": summary[0]["summary_text"]})
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
+