Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def load_model(name):
|
|
17 |
|
18 |
return tokenizer, model.to(device)
|
19 |
|
20 |
-
tokenizer1, model1 = load_model("
|
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 (
|
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 |
],
|