Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
| 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 (
|
| 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",
|