tree3po commited on
Commit
dfcc13c
1 Parent(s): a3833ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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(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):
 
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):