Updated ais to right models

app.py

@@ -9,8 +9,8 @@ def load_model_and_tokenizer(model_name):
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return model, tokenizer
 
-model_8b, tokenizer_8b = load_model_and_tokenizer("huggyllama/llama-3.1-8b")
-model_8b_instruct, tokenizer_8b_instruct = load_model_and_tokenizer("huggyllama/llama-3.1-8b-instruct")
+model_8b, tokenizer_8b = load_model_and_tokenizer("meta-llama/Meta-Llama-3.1-8B")
+model_8b_instruct, tokenizer_8b_instruct = load_model_and_tokenizer("meta-llama/Meta-Llama-3.1-8B-Instruct")
 
 def generate_text(model, tokenizer, prompt, max_length=100):
     inputs = tokenizer(prompt, return_tensors="pt")