Omnibus commited on
Commit
11454fa
1 Parent(s): cde580d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -11
app.py CHANGED
@@ -8,24 +8,20 @@ models=[
8
  def load_models():
9
  loaded_model=[]
10
  model_details=[]
11
- for model in models:
 
12
  try:
13
- loaded_model.append(gr.load(f'models/{model}'))
14
- this = api.list_models(search=model.split("/")[1],limit=100,cardData=True)
15
- for i,m in enumerate(this):
16
- model_details.append(m)
17
  except Exception as e:
18
- loaded_model.append({"ERROR":e})
19
  try:
20
- this = api.list_models(search=model.split("/")[1],limit=100,cardData=True)
21
- for i,m in enumerate(this):
22
- model_details.append(m)
23
  except Exception as ee:
24
- model_details.append({"ERROR":ee})
25
  return loaded_model, model_details
26
  with gr.Blocks() as app:
27
  with gr.Row():
28
  models_out=gr.JSON(label="Gradio Details")
29
- details=gr.JSON(label="API Details")
30
  app.load(load_models,None,[models_out,details])
31
  app.launch()
 
8
  def load_models():
9
  loaded_model=[]
10
  model_details=[]
11
+ similar_models = api.list_models(search=models[0].split("/")[1],limit=100,cardData=True)
12
+ for model in similar_models:
13
  try:
14
+ loaded_model.append(gr.load(f'models/{model.id}'))
 
 
 
15
  except Exception as e:
16
+ loaded_model.append({"MODEL":model.id,"ERROR":e})
17
  try:
18
+ model_details.append(model)
 
 
19
  except Exception as ee:
20
+ model_details.append({"MODEL":model.id,"ERROR":ee})
21
  return loaded_model, model_details
22
  with gr.Blocks() as app:
23
  with gr.Row():
24
  models_out=gr.JSON(label="Gradio Details")
25
+ details=gr.JSON(label="Hub Details")
26
  app.load(load_models,None,[models_out,details])
27
  app.launch()