Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 # Load model and tokenizer
-model_name = "llama-2-7b-chat"
+model_name = "meta-llama/Llama-2-7b-chat-hf"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)