Update app.py

app.py

@@ -10,7 +10,7 @@ from huggingface_hub import login
 import spaces
 
 # 🔐 Login (Hugging Face token should be set as HF_TOKEN env variable)
-login(token=os.environ.get("HF_TOKEN"))
+login(token=os.environ.get("token"))
 
 # ✅ Check for GPU
 if not torch.cuda.is_available():
@@ -19,12 +19,12 @@ print(f"✅ Using GPU: {torch.cuda.get_device_name(0)}")
 
 # Model setup
 model_id = "mistralai/Mistral-7B-Instruct-v0.2"
-tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ.get("HF_TOKEN"))
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ.get("token"))
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map="auto",
     torch_dtype=torch.float16,
-    token=os.environ.get("HF_TOKEN"),
+    token=os.environ.get("token"),
     trust_remote_code=True
 )
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)