Update app/main.py

app/main.py

@@ -1,18 +1,26 @@
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from app.api import ask, rewrite, home
-
-app = FastAPI()
-
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  # Allows all origins
-    allow_credentials=True,
-    allow_methods=["*"],  # Allows all methods
-    allow_headers=["*"],  # Allows all headers
-)
-
-app.include_router(ask.router)
-app.include_router(rewrite.router)
-app.include_router(home.router)
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from app.api import ask, rewrite, home
+import os
+
+os.environ["HF_HOME"] = "./cache"
+os.environ["TRANSFORMERS_CACHE"] = "./cache"
+os.environ["TORCH_HOME"] = "./cache"
+
+# Tạo folder cache nếu chưa tồn tại
+os.makedirs("./cache", exist_ok=True)
+
+app = FastAPI()
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
+app.include_router(ask.router)
+app.include_router(rewrite.router)
+app.include_router(home.router)