File size: 1,501 Bytes
c9bb6b3 639ed59 403c2be c9bb6b3 403c2be 639ed59 cde580d 639ed59 6f991e0 403c2be 6f991e0 3a82911 6f991e0 11454fa b72a049 403c2be 4062274 c374be7 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 41 42 43 44 45 |
import sys
import gradio as gr
from huggingface_hub import HfApi, ModelFilter, list_models
from subprocess import run
#result = run(['script.py'], capture_output=True)
#previous_output = result.stdout
api = HfApi()
models=[
"xai-org/grok-1"
]
def load_models(model_in):
loaded_model=[]
model_details=[]
if "/" in model_in:
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_path=f'models/{model.id}'
this = gr.load(model_path,cache_examples=False).stdout
print(this)
#loaded_model.append(gr.load(model_path,cache_examples=False).stdout)
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()
|