Sanchit2207 commited on
Commit
39ad3e5
·
verified ·
1 Parent(s): c4a4b0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ def load_model(name):
17
  return tokenizer, model.to(device)
18
 
19
  tokenizer1, model1 = load_model("Gensyn/Qwen2.5-0.5B-Instruct")
20
- tokenizer2, model2 = load_model("google/flan-t5-small")
21
  tokenizer3, model3 = load_model("microsoft/phi-1_5")
22
 
23
  def generate_response(model, tokenizer, prompt):
@@ -50,7 +50,7 @@ interface = gr.Interface(
50
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
51
  outputs=[
52
  gr.Textbox(label="Agent 1 (Gensyn/Qwen2.5-0.5B-Instruct)"),
53
- gr.Textbox(label="Agent 2 (google/flan-t5-small)"),
54
  gr.Textbox(label="Agent 3 (microsoft/phi-1_5)")
55
  ],
56
  title="3-Agent AI Chatbot",
 
17
  return tokenizer, model.to(device)
18
 
19
  tokenizer1, model1 = load_model("Gensyn/Qwen2.5-0.5B-Instruct")
20
+ tokenizer2, model2 = load_model("tiiuae/falcon-rw-1b")
21
  tokenizer3, model3 = load_model("microsoft/phi-1_5")
22
 
23
  def generate_response(model, tokenizer, prompt):
 
50
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
51
  outputs=[
52
  gr.Textbox(label="Agent 1 (Gensyn/Qwen2.5-0.5B-Instruct)"),
53
+ gr.Textbox(label="Agent 2 (tiiuae/falcon-rw-1b)"),
54
  gr.Textbox(label="Agent 3 (microsoft/phi-1_5)")
55
  ],
56
  title="3-Agent AI Chatbot",