Omnibus commited on
Commit
403c2be
1 Parent(s): 639ed59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -1,14 +1,25 @@
1
  import gradio as gr
2
- #from models import models
 
3
 
4
  models=[
5
  "xai-org/grok-1",
6
  ]
7
- loaded_model=[]
8
- for i,model in enumerate(models):
9
- try:
10
- loaded_model.append(gr.load(f'models/{model}'))
11
- except Exception as e:
12
- print(e)
13
- pass
14
- print (loaded_model)
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from huggingface_hub import HfApi, ModelFilter, list_models
3
+ api = HfApi()
4
 
5
  models=[
6
  "xai-org/grok-1",
7
  ]
8
+ def load_models():
9
+ loaded_model=[]
10
+ model_details=[]
11
+ for i,model in enumerate(models):
12
+ try:
13
+ loaded_model.append(gr.load(f'models/{model}'))
14
+ this = api.list_models(search=model,limit=limit,cardData=True)
15
+ model_details.append(this)
16
+ except Exception as e:
17
+ print(e)
18
+ pass
19
+ return loaded_models, model_details
20
+ with gr.Blocks() as app:
21
+ with gr.Row():
22
+ models=gr.JSON()
23
+ details=gr.JSON()
24
+ app.load(load_models,None,[models,details])
25
+ app.launch()