Update app.py

app.py

@@ -14,11 +14,18 @@ logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# Load T5 model and tokenizer (done once at startup)
+# Set Hugging Face cache directory (matches Dockerfile)
+os.environ["HF_HOME"] = "/app/hf_cache"
+
+# Load T5 model and tokenizer
 logger.info("Loading T5-Base model...")
-tokenizer = T5Tokenizer.from_pretrained("t5-base")
-model = T5ForConditionalGeneration.from_pretrained("t5-base")
-logger.info("T5-Base model loaded successfully.")
+try:
+    tokenizer = T5Tokenizer.from_pretrained("t5-base")
+    model = T5ForConditionalGeneration.from_pretrained("t5-base")
+    logger.info("T5-Base model loaded successfully.")
+except Exception as e:
+    logger.error(f"Failed to load T5-Base: {str(e)}")
+    raise
 
 ALLOWED_EXTENSIONS = {"pdf", "docx", "pptx", "txt"}
 
@@ -28,7 +35,6 @@ def allowed_file(filename):
 def summarize_text(text, max_length=150, min_length=30):
     """Summarize text using T5-Base."""
     try:
-        # Prepend "summarize: " as required by T5
         input_text = "summarize: " + text
         inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
         summary_ids = model.generate(