Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
-
+hf_token = os.getenv('HF_API_TOKEN')
 # Load the Llama 3.1 model and tokenizer
 model_name = "meta-llama/Meta-Llama-3.1-8B"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_name, token= hf_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto",token = hf_token)
 
 # Streamlit app interface
 st.title("Llama 3.1 Text Generator")