Update app.py

app.py

@@ -1,11 +1,12 @@
 import os
 from transformers import pipeline
 
-# Set custom cache directory to avoid permission issues
+# Set a custom cache directory to avoid permission issues
 os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+os.environ["HF_HOME"] = "/app/cache"
 
-# Load the model with TensorFlow weights if PyTorch version is unavailable
-summarizer = pipeline("summarization", model="t5-base", from_tf=True)
+# Load model (force PyTorch)
+summarizer = pipeline("summarization", model="t5-base", framework="pt")
 
 from flask import Flask, request, jsonify