Update app.py

app.py

@@ -6,9 +6,13 @@ import vertexai
 from vertexai.preview.vision_models import ImageGenerationModel
 
 # --- 1. Authentication and Initialization ---
-# This section handles secure authentication with Google Cloud.
-# It expects the GCP credentials to be stored in a Hugging Face Secret
-# named "GOOGLE_APPLICATION_CREDENTIALS_JSON".
+# Part A: Hugging Face Hub Authentication
+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. Hub-related features may be disabled.")
 
 # Check if the secret is available
 if "GOOGLE_APPLICATION_CREDENTIALS_JSON" in os.environ: