Update app.py

app.py

@@ -1,12 +1,16 @@
 import os
 from transformers import pipeline
 
-# Set a custom cache directory to avoid permission issues
-os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
-os.environ["HF_HOME"] = "/app/cache"
+# Set a writable cache directory inside Hugging Face Space
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/cache"
+
+# Ensure the cache directory exists
+os.makedirs("/tmp/cache", exist_ok=True)
+
+# Load model
+summarizer = pipeline("summarization", model="t5-base", from_tf=True)
+print(summarizer("This is a long text that needs summarization.", max_length=50, min_length=10, do_sample=False))
 
-# Load model (force PyTorch)
-summarizer = pipeline("summarization", model="t5-base", framework="pt")
 
 from flask import Flask, request, jsonify