Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
-# Load Mistral-7B model
-model_name = "mistralai/Mistral-7B-Instruct-v0.1"
+# Load Llama-2 model
+model_name = "meta-llama/Llama-2-7b-chat-hf"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
 
@@ -27,7 +27,7 @@ demo = gr.Interface(
     fn=chat,
     inputs=["dropdown", "text"],
     outputs="text",
-    title="Drapel – Chat with AI Celebrities",
+    title="Chat with AI Celebrities",
     description="Select a character and chat with their AI version.",
     examples=[["Albert Einstein", "What is relativity?"], ["Cristiano Ronaldo", "How do you stay motivated?"]]
 )