Omnibus commited on
Commit
f47d997
·
verified ·
1 Parent(s): 2be8239

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -9,13 +9,16 @@ def model_explorer(search,limit=100,task='text-generation'):
9
  if not search:
10
  search=None
11
  out_box=[]
 
12
  filt = ModelFilter(task="text-generation")
13
 
14
  this = api.list_models(search=search,limit=limit,filter=filt,cardData=True)
15
-
16
  for i,mod in enumerate(this):
 
17
  if mod.gated=='manual':
18
  out_box.append(mod)
 
19
  '''
20
  print(dir(mod))
21
  print(mod.id)
@@ -24,7 +27,8 @@ def model_explorer(search,limit=100,task='text-generation'):
24
  print(mod.pipeline_tag)
25
  out = mod.id
26
  '''
27
- return out_box
 
28
 
29
  def tasks_json():
30
 
@@ -88,6 +92,7 @@ with gr.Blocks() as app:
88
  search=gr.Textbox()
89
  limit=gr.Slider(minimum=0,maximum=10000,value=100)
90
  btn=gr.Button()
 
91
  models_json=gr.JSON()
92
- btn.click(model_explorer,[search,limit],models_json)
93
  app.launch()
 
9
  if not search:
10
  search=None
11
  out_box=[]
12
+ name_box=[]
13
  filt = ModelFilter(task="text-generation")
14
 
15
  this = api.list_models(search=search,limit=limit,filter=filt,cardData=True)
16
+ cnt=0
17
  for i,mod in enumerate(this):
18
+ cnt+=1
19
  if mod.gated=='manual':
20
  out_box.append(mod)
21
+ name_box.append(mod.id)
22
  '''
23
  print(dir(mod))
24
  print(mod.id)
 
27
  print(mod.pipeline_tag)
28
  out = mod.id
29
  '''
30
+ print(cnt)
31
+ return out_box,name_box
32
 
33
  def tasks_json():
34
 
 
92
  search=gr.Textbox()
93
  limit=gr.Slider(minimum=0,maximum=10000,value=100)
94
  btn=gr.Button()
95
+ names_json=gr.JSON()
96
  models_json=gr.JSON()
97
+ btn.click(model_explorer,[search,limit],[models_json,names_json])
98
  app.launch()