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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ def load_model(name):
18
  return tokenizer, model.to(device)
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
@@ -52,7 +52,7 @@ interface = gr.Interface(
52
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
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",
 
18
  return tokenizer, model.to(device)
19
 
20
  tokenizer1, model1 = load_model("Gensyn/Qwen2.5-0.5B-Instruct")
21
+ tokenizer2, model2 = load_model("deepset/roberta-base-squad2")
22
  tokenizer3, model3 = load_model("microsoft/phi-1_5")
23
 
24
  # Generation function
 
52
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
53
  outputs=[
54
  gr.Textbox(label="Agent 1 (Gensyn/Qwen2.5-0.5B-Instruct)"),
55
+ gr.Textbox(label="Agent 2 (deepset/roberta-base-squad2)"),
56
  gr.Textbox(label="Agent 3 (microsoft/phi-1_5)")
57
  ],
58
  title="3-Agent AI Chatbot",