Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+from flask import Flask, request, jsonify
 from transformers import pipeline
 
 # Ensure HF doesn't request a token
@@ -10,8 +11,25 @@ os.environ["HF_HUB_OFFLINE"] = "0"
 # Load model
 summarizer = pipeline("summarization", model="t5-base")
 
-# Ensure script runs properly
+app = Flask(__name__)
+
+@app.route("/")
+def home():
+    return "Summarization API is running!"
+
+@app.route("/summarize", methods=["POST"])
+def summarize_text():
+    data = request.get_json()
+    text = data.get("text", "")
+    max_length = data.get("max_length", 50)
+    min_length = data.get("min_length", 10)
+
+    if not text:
+        return jsonify({"error": "No text provided"}), 400
+
+    summary = summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
+    return jsonify(summary)
+
 if __name__ == "__main__":
-    print("Application started successfully!")
-    result = summarizer("This is a long text that needs summarization.", max_length=50, min_length=10, do_sample=False)
-    print(result)
+    print("🚀 API is running on port 7860")
+    app.run(host="0.0.0.0", port=7860)