Update app.py

app.py

@@ -4,6 +4,13 @@ from typing import Union
 from huggingface_hub import whoami
 is_spaces = True if os.environ.get("SPACE_ID") else False
 
+from huggingface_hub import login
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+else:
+    raise ValueError("HF_TOKEN environment variable not found!")
+
 if is_spaces:
     subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
     import spaces