Use pipeline

app.py

@@ -6,10 +6,9 @@ hf_api_token = st.secrets["HUGGINGFACE_API_TOKEN"]
 
 # Load the model and tokenizer using the API token
 model_name = "meta-llama/Meta-Llama-3.1-8B"
-tokenizer = pipeline('text-generation', model_name, token=hf_api_token)
 
 # Create a text generation pipeline
-generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+generator = pipeline("text-generation", model=model, token=hf_api_token)
 
 # Streamlit UI
 st.title("LLaMA 3.1-405B Model Text Generation")