Update app.py

app.py

@@ -18,7 +18,7 @@ def load_model(name):
     return tokenizer, model.to(device)
 
 tokenizer1, model1 = load_model("EleutherAI/gpt-neo-125M")
-tokenizer2, model2 = load_model("distilgpt2")
+tokenizer2, model2 = load_model("facebook/opt-125m")
 tokenizer3, model3 = load_model("gpt2")
 
 # Generation function
@@ -52,7 +52,7 @@ interface = gr.Interface(
     inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
     outputs=[
         gr.Textbox(label="Agent 1 (EleutherAI/gpt-neo-125M)"),
-        gr.Textbox(label="Agent 2 (DistilGPT-2)"),
+        gr.Textbox(label="Agent 2 (facebook/opt-125m)"),
         gr.Textbox(label="Agent 3 (GPT-2)")
     ],
     title="3-Agent AI Chatbot",