Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
|
|
|
| 2 |
import random
|
| 3 |
models=[
|
| 4 |
"facebook/MobileLLM-125M",
|
|
@@ -18,7 +20,7 @@ def load_models(inp,new_models):
|
|
| 18 |
#print(new_models[inp[0]])
|
| 19 |
client_z.clear()
|
| 20 |
for z,ea in enumerate(inp):
|
| 21 |
-
client_z.append(
|
| 22 |
out_box[z]=(gr.update(label=new_models[inp[z]]))
|
| 23 |
return out_box[0],out_box[1],out_box[2],out_box[3]
|
| 24 |
def format_prompt_default(message, history):
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
from gradio_client import Client
|
| 3 |
+
|
| 4 |
import random
|
| 5 |
models=[
|
| 6 |
"facebook/MobileLLM-125M",
|
|
|
|
| 20 |
#print(new_models[inp[0]])
|
| 21 |
client_z.clear()
|
| 22 |
for z,ea in enumerate(inp):
|
| 23 |
+
client_z.append(Client("models/" + new_models[inp[z]]))
|
| 24 |
out_box[z]=(gr.update(label=new_models[inp[z]]))
|
| 25 |
return out_box[0],out_box[1],out_box[2],out_box[3]
|
| 26 |
def format_prompt_default(message, history):
|