Update app.py

app.py

@@ -3,9 +3,13 @@ import torch
 from peft import PeftModel
 import gradio as gr
 from huggingface_hub import login
+import os
 
-# Log in with the secret token (stored in Hugging Face Secrets)
-login(token="${HF_TOKEN}")
+# Retrieve the token from the environment variable and log in
+hf_token = os.environ.get("HF_TOKEN")
+if hf_token is None:
+    raise ValueError("HF_TOKEN environment variable not found. Please check your Space secrets.")
+login(token=hf_token)
 
 # Define model paths
 base_model_name = "meta-llama/Llama-3.2-3B-Instruct"