Update app.py

app.py

@@ -1,10 +1,14 @@
 import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
+from huggingface_hub import login
 
 # Load the Hugging Face API token from environment variables
 hf_token = os.getenv("HF_TOKEN")
 
+# Login to huggingface_hub using the token
+login(token=hf_token)
+
 # Model name
 model_name = "meta-llama/Llama-2-7b-chat-hf"