Update app/main.py

app/main.py

@@ -1,18 +1,21 @@
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from app.routes import summarize
-
-app = FastAPI()
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-@app.get("/")
-def read_root():
-    return {"message": "AutoTLDR backend is running"}
-
-app.include_router(summarize.router)
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from app.routes import summarize
+import os
+
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/")
+def read_root():
+    return {"message": "AutoTLDR backend is running"}
+
+app.include_router(summarize.router)