Update app.py

app.py

@@ -66,31 +66,29 @@ if __name__ == "__main__":
     demo.launch()
 
 import os
-
-api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 import torch
 
-# Load the Hugging Face API token securely
+# Load Hugging Face API token securely
 api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
-if api_token is None:
-    raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set. Please set it in your environment variables.")
+if not api_token:
+    raise ValueError("❌ ERROR: Hugging Face API token is not set. Please set it as an environment variable.")
 
 # Define model names
 base_model_name = "unsloth/qwen2.5-math-7b-bnb-4bit"
 peft_model_name = "Hrushi02/Root_Math"
 
-# Load base model securely
+# Load base model with authentication
 base_model = AutoModelForCausalLM.from_pretrained(
     base_model_name, 
     torch_dtype=torch.float16, 
     device_map="auto",
-    token=api_token
+    token=api_token  # ✅ Pass token here
 )
 
-# Load PEFT fine-tuned model
+# Load fine-tuned model
 model = PeftModel.from_pretrained(base_model, peft_model_name, token=api_token)
 
 # Load tokenizer
@@ -98,3 +96,4 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_name, token=api_token)
 
 
 
+