Update app.py

app.py

@@ -14,13 +14,13 @@ import sentencepiece
 import huggingface_hub
 from transformers import TrainingArguments, Trainer
 
-# Retrieve HF_TOKEN from Hugging Face Space secrets
-HF_TOKEN = os.getenv("HF_TOKEN:levi put token here")  # Token expected as env variable 'HF_TOKEN'
-if not HF_TOKEN:
-    raise ValueError("HF_TOKEN not found in environment variables. Please set it in Hugging Face Space secrets under 'Settings' > 'Secrets'.")
+# Retrieve the token from Hugging Face Space secrets
+llama = os.getenv("llama:levi put token here")  # Token expected as env variable 'llama'
+if not llama:
+    raise ValueError("llama token not found in environment variables. Please set it in Hugging Face Space secrets under 'Settings' > 'Secrets' as 'llama'.")
 
 # Authenticate with Hugging Face
-huggingface_hub.login(token=HF_TOKEN)
+huggingface_hub.login(token=llama)
 
 # Model setup
 MODEL_ID = "meta-llama/Llama-2-7b-hf"
@@ -143,7 +143,7 @@ def train_ui(files):
         return "Training completed! Model saved to ./fine_tuned_llama_healthcare"
 
     except Exception as e:
-        return f"Error: {str(e)}. Please check file format, dependencies, or HF_TOKEN."
+        return f"Error: {str(e)}. Please check file format, dependencies, or the llama token."
 
 # Gradio UI
 with gr.Blocks(title="Healthcare Fraud Detection Fine-Tuning") as demo: