Update main.py

main.py

@@ -1,3 +1,8 @@
+import os
+
+# Set cache directory for Hugging Face
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+
 from fastapi import FastAPI
 from transformers import pipeline
 import uvicorn
@@ -17,5 +22,7 @@ def summarize(text: str):
     summary = summarizer(text, max_length=150, min_length=50, do_sample=False)
     return {"summary": summary[0]['summary_text']}
 
+# Run Uvicorn server
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)
+