Update app.py

app.py

@@ -1,5 +1,14 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import login
+import os
+
+# Retrieve the Hugging Face token from the Space secrets
+token = os.getenv("HF_TOKEN")
+
+# Log in using the token
+login(token=token)
+
 
 # Load model and tokenizer
 model_name = "meta-llama/Llama-3.2-1B"  # Replace with the correct model name if necessary