File size: 1,027 Bytes
639ed59
403c2be
 
639ed59
 
d184170
639ed59
403c2be
 
 
43e3edc
403c2be
004ef5c
c287b8f
03202d1
 
403c2be
9af1fae
 
c287b8f
03202d1
 
9af1fae
 
168f093
403c2be
 
03202d1
68d9ea2
f4505b9
403c2be
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import gradio as gr
from huggingface_hub import HfApi, ModelFilter, list_models
api = HfApi()

models=[
    "grok-1"
]
def load_models():
    loaded_model=[]
    model_details=[]
    for model in models:
        try:
            loaded_model.append(gr.load(f'models/xai-org/{model}'))
            this = api.list_models(search=model,limit=100,cardData=True)
            for i,m in enumerate(this):
                model_details.append(m)
        except Exception as e:
            loaded_model.append({"ERROR":e})
            try:
                this = api.list_models(search=model,limit=100,cardData=True)
                for i,m in enumerate(this):
                    model_details.append(m)
            except Exception as ee:
                model_details.append({"ERROR":ee})
    return loaded_model, model_details
with gr.Blocks() as app:
    with gr.Row():
        models_out=gr.JSON(label="Gradio Details")
        details=gr.JSON(label="API Details")
    app.load(load_models,None,[models_out,details])
app.launch()