Update app.py

app.py

@@ -4,11 +4,11 @@ import torch
 import json
 
 title = "AI ChatBot"
-description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
+description = "A State-of-the-Art Large-scale Pretrained Response generation model (GEMMA)"
 examples = [["How are you?"]]
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+tokenizer = AutoTokenizer.from_pretrained("your-gemma-model-name")
+model = AutoModelForCausalLM.from_pretrained("your-gemma-model-name")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
@@ -62,7 +62,7 @@ gr.Interface(
     title=title,
     description=description,
     examples=examples,
-    inputs=["text", "state"],  # Changed input from "state" to "text"
-    outputs=["chatbot", "state"],  # Changed output to match the tuple return type
+    inputs=["text"],
+    outputs=["text", "state"],
     theme="finlaymacklon/boxy_violet"
-).launch()
+).launch()