Update app.py

app.py

@@ -1,8 +1,11 @@
-from transformers import pipeline
 import os
-os.environ["TRANSFORMERS_CACHE"] = "/app/cache"  # Set custom cache directory
+from transformers import pipeline
 
-summarizer = pipeline("summarization", model="t5-base", framework="pt")  # Force PyTorch
+# Set custom cache directory to avoid permission issues
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
+# Load the model with TensorFlow weights if PyTorch version is unavailable
+summarizer = pipeline("summarization", model="t5-base", from_tf=True)
 
 from flask import Flask, request, jsonify
 
@@ -17,4 +20,3 @@ def summarize():
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
-