Update main.py

main.py

@@ -14,6 +14,10 @@ import uvicorn
 from starlette.concurrency import run_in_threadpool
 import subprocess, sys
 
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+login(token=HF_TOKEN)
+
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 
 DATA_DIR = os.environ.get("DATA_DIR", "/app/data")