Update app.py

app.py

@@ -3,16 +3,18 @@ from transformers import pipeline
 import torch
 import os
 
+# Set HF token if available in environment (provided via Hugging Face Spaces Secrets)
+# This is CRITICAL to access gated models like Mistral-7B-Instruct-v0.2
+from huggingface_hub import login
+login(os.getenv("HUGGINGFACEHUB_API_TOKEN"))  # automatically handles secrets
+
 # Use safe float32 for CPU compatibility
 torch_dtype = torch.float32
 
 # Configure cache directory
 os.environ['HF_HOME'] = '/tmp/cache'
 
-# Optional: print token to verify it's passed at runtime
-# print("HF Token:", os.environ.get("HUGGINGFACEHUB_API_TOKEN"))
-
-# Load the model pipeline with auth token picked up automatically
+# Load the model pipeline
 generator = pipeline(
     "text-generation",
     model="mistralai/Mistral-7B-Instruct-v0.2",