Sanchit2207 commited on
Commit
084a079
·
verified ·
1 Parent(s): b52c793

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ def load_model(name):
19
 
20
  tokenizer1, model1 = load_model("Gensyn/Qwen2.5-0.5B-Instruct")
21
  tokenizer2, model2 = load_model("facebook/opt-125m")
22
- tokenizer3, model3 = load_model("mistralai/Mistral-7B-Instruct-v0.2")
23
 
24
  # Generation function
25
  def generate_response(model, tokenizer, prompt):
@@ -53,7 +53,7 @@ interface = gr.Interface(
53
  outputs=[
54
  gr.Textbox(label="Agent 1 (Gensyn/Qwen2.5-0.5B-Instruct)"),
55
  gr.Textbox(label="Agent 2 (facebook/opt-125m)"),
56
- gr.Textbox(label="Agent 3 (mistralai/Mistral-7B-Instruct-v0.2)")
57
  ],
58
  title="3-Agent AI Chatbot",
59
  description="Three GPT-style agents respond to your input in parallel!"
 
19
 
20
  tokenizer1, model1 = load_model("Gensyn/Qwen2.5-0.5B-Instruct")
21
  tokenizer2, model2 = load_model("facebook/opt-125m")
22
+ tokenizer3, model3 = load_model("microsoft/phi-1_5")
23
 
24
  # Generation function
25
  def generate_response(model, tokenizer, prompt):
 
53
  outputs=[
54
  gr.Textbox(label="Agent 1 (Gensyn/Qwen2.5-0.5B-Instruct)"),
55
  gr.Textbox(label="Agent 2 (facebook/opt-125m)"),
56
+ gr.Textbox(label="Agent 3 (microsoft/phi-1_5)")
57
  ],
58
  title="3-Agent AI Chatbot",
59
  description="Three GPT-style agents respond to your input in parallel!"