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