using llama 3

app.py

@@ -2,8 +2,12 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 # Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+#tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")#
+#model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+
+
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
 
 # Create text generation pipeline
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)