Update app.py

app.py

@@ -22,18 +22,33 @@ print("=" * 50)
 # ------------------------------------------------------
 # 🔹 STEP 1: Authentication for HF Spaces
 # ------------------------------------------------------
-hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+# Check both possible token names (order matters!)
+hf_token = os.getenv("HUGGINGFACE_HUB_TOKEN") or os.getenv("HF_TOKEN")
 
-if hf_token:
-    print("✅ HF Token found, logging in...")
-    login(token=hf_token)
-else:
-    print("⚠️ No HF_TOKEN found - attempting to load model without authentication")
+if not hf_token:
+    raise ValueError(
+        "❌ No token found!\n"
+        "Please add your Hugging Face token in Space Settings → Repository secrets.\n"
+        "Name it either: HUGGINGFACE_HUB_TOKEN or HF_TOKEN\n"
+        "Create a token at: https://huggingface.co/settings/tokens"
+    )
+
+print(f"✅ HF Token found (length: {len(hf_token)})")
+print("🔐 Logging in to Hugging Face...")
+login(token=hf_token)
+print("✅ Login successful!")
 
 # ------------------------------------------------------
 # 🔹 STEP 2: Load model and tokenizer
 # ------------------------------------------------------
-model_id = "google/medgemma-27b-text-it"
+# Option 1: MedGemma (requires access request)
+# model_id = "google/medgemma-27b-text-it"
+
+# Option 2: Regular Gemma (no access needed, works immediately)
+model_id = "google/gemma-2-9b-it"  # Smaller, faster, no access required
+
+# Option 3: Mistral (medical fine-tuned alternative)
+# model_id = "mistralai/Mistral-7B-Instruct-v0.3"
 
 print("🔄 Loading tokenizer...")
 try: