Update app.py with another model

app.py

@@ -1,9 +1,9 @@
-from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
@@ -26,8 +26,8 @@ demo = gr.Interface(
     fn=generate_response,
     inputs=gr.Textbox(lines=5, placeholder="Talk to the model..."),
     outputs=gr.Textbox(label="Response", lines=10),
-    title="Chat with Llama 2 7B",
-    description="Open-access LLM demo."
+    title="GPT-2 Chatbot",
+    description="Chat with GPT-2 model."
 )
 
 if __name__ == "__main__":