Update app.py

app.py

@@ -1,27 +1,7 @@
-import os
 from transformers import pipeline
 
-# Set a writable cache directory inside Hugging Face Space
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/cache"
+# Load the summarization model (ensure from_tf is removed)
+summarizer = pipeline("summarization", model="t5-base")
 
-# Ensure the cache directory exists
-os.makedirs("/tmp/cache", exist_ok=True)
-
-# Load model
-summarizer = pipeline("summarization", model="t5-base", from_tf=True)
+# Example usage
 print(summarizer("This is a long text that needs summarization.", max_length=50, min_length=10, do_sample=False))
-
-
-from flask import Flask, request, jsonify
-
-app = Flask(__name__)
-
-@app.route("/summarize", methods=["POST"])
-def summarize():
-    data = request.json
-    text = data.get("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)