Update app.py

app.py

@@ -8,19 +8,29 @@ import time
 import base64
 import google.auth
 import google.auth.transport.requests
-from huggingface_hub import login
+import spaces
+from huggingface_hub import login, hf_hub_download
 
 # --- 1. Configuration and Authentication (Unchanged) ---
 GCP_PROJECT_ID = os.environ.get("GCP_PROJECT_ID")
 GCP_LOCATION = os.environ.get("GCP_LOCATION")
 
 # --- Authentication and Sanity Checks Block (Unchanged) ---
-hf_token = os.environ.get("HF_TOKEN")
-if hf_token:
-    print("Hugging Face token found. Logging in.")
-    login(token=hf_token)
-else:
-    print("WARNING: Hugging Face token ('HF_TOKEN') not found.")
+
+
+# Set up HF_TOKEN for Hugging Face Hub authentication
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise EnvironmentError(
+        "HF_TOKEN environment variable not found. Please set it in the Hugging Face Space secrets"
+    )
+
+try:
+    login(token=HF_TOKEN)
+except Exception as e:
+    raise EnvironmentError(
+        f"Failed to authenticate with HF_TOKEN. Ensure the token has read access."
+    )
 
 creds_json_str = os.environ.get("GOOGLE_APPLICATION_CREDENTIALS_JSON")