Llama 8B

app.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 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-405B"
+model_name = "meta-llama/Meta-Llama-3.1-8B"
 tokenizer = pipeline('text-generation', model_name, token=hf_api_token)
 
 # Create a text generation pipeline