Sanchit2207 commited on
Commit
a905023
·
verified ·
1 Parent(s): 77b0f0f

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
 
18
  return tokenizer, model.to(device)
19
 
20
- tokenizer1, model1 = load_model("EleutherAI/gpt-neo-125M")
21
  tokenizer2, model2 = load_model("facebook/opt-125m")
22
  tokenizer3, model3 = load_model("gpt2")
23
 
@@ -51,7 +51,7 @@ interface = gr.Interface(
51
  fn=multi_agent_chat,
52
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
53
  outputs=[
54
- gr.Textbox(label="Agent 1 (EleutherAI/gpt-neo-125M)"),
55
  gr.Textbox(label="Agent 2 (facebook/opt-125m)"),
56
  gr.Textbox(label="Agent 3 (GPT-2)")
57
  ],
 
17
 
18
  return tokenizer, model.to(device)
19
 
20
+ tokenizer1, model1 = load_model("TheBloke/phi-2-GGUF")
21
  tokenizer2, model2 = load_model("facebook/opt-125m")
22
  tokenizer3, model3 = load_model("gpt2")
23
 
 
51
  fn=multi_agent_chat,
52
  inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
53
  outputs=[
54
+ gr.Textbox(label="Agent 1 (TheBloke/phi-2-GGUF)"),
55
  gr.Textbox(label="Agent 2 (facebook/opt-125m)"),
56
  gr.Textbox(label="Agent 3 (GPT-2)")
57
  ],