Spaces:
Running
Running
File size: 1,356 Bytes
639ed59 403c2be 639ed59 cde580d 639ed59 6f991e0 403c2be 6f991e0 11454fa 403c2be 8522a2b 75cd5da 403c2be 11454fa b798126 168f093 403c2be 6f991e0 1155e20 403c2be 03202d1 11454fa 876a6c2 570ccce 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 33 34 35 36 37 38 39 40 |
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.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()
|