app.py

import gradio as gr
from huggingface_hub import HfApi, ModelFilter, list_models
api = HfApi()

models=[
    "xai-org/grok-1"
]
def load_models(model_in):
    loaded_model=[]
    model_details=[]
    
    if "/" in models:
        similar_models = api.list_models(search=model_in.split("/")[1],limit=100,cardData=True)
    else:
        similar_models = api.list_models(search=model_in,limit=100,cardData=True)
    for model in similar_models:
        try:
            model_load=gr.Interface.load(f'models/{model.id}')
            print(dir(model_load))
            print(model_load)
            #out_test=model_load("hello?")
            loaded_model.append(model_load)
        except Exception as e:
            loaded_model.append({"MODEL":model.id,"ERROR":e})
        try:
            model_details.append(model)
        except Exception as ee:
            model_details.append({"MODEL":model.id,"ERROR":ee})
    return loaded_model, model_details
with gr.Blocks() as app:
    with gr.Row():
        model_name=gr.Textbox(label="Model", value=models[0], placeholder=models[0])
        load_btn=gr.Button("Load")
    with gr.Row():
        models_out=gr.JSON(label="Gradio Details")
        details=gr.JSON(label="Hub Details")
    app.load(load_models,model_name,[models_out,details])
    load_btn.click(load_models,model_name,[models_out,details])
app.launch()