Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
@@ -11,6 +12,13 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 # Authenticate with the access token
 login("HF_ACCESS_TOKEN")
 
+# Retrieve token securely
+token = os.getenv("HF_ACCESS_TOKEN")
+if token is None:
+    raise ValueError("HF_ACCESS_TOKEN not found. Did you set it in the Secrets?")
+login(token)
+
+
 model_name = "ayeshaNoor1/Llama_finetunedModel-1b"
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)