Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,11 +5,9 @@ api = HfApi()
|
|
| 5 |
models=[
|
| 6 |
"xai-org/grok-1"
|
| 7 |
]
|
| 8 |
-
def list_files():
|
| 9 |
-
files = api.list_repo_files(repo_id=
|
| 10 |
-
|
| 11 |
-
list_files()
|
| 12 |
-
|
| 13 |
|
| 14 |
def load_models(model_in):
|
| 15 |
loaded_model=[]
|
|
@@ -36,9 +34,12 @@ with gr.Blocks() as app:
|
|
| 36 |
with gr.Row():
|
| 37 |
model_name=gr.Textbox(label="Model", value=models[0], placeholder=models[0])
|
| 38 |
load_btn=gr.Button("Load")
|
|
|
|
|
|
|
| 39 |
with gr.Row():
|
| 40 |
models_out=gr.JSON(label="Gradio Details")
|
| 41 |
details=gr.JSON(label="Hub Details")
|
|
|
|
| 42 |
#app.load(load_models,model_name,[models_out,details])
|
| 43 |
load_btn.click(load_models,model_name,[models_out,details])
|
| 44 |
app.launch()
|
|
|
|
| 5 |
models=[
|
| 6 |
"xai-org/grok-1"
|
| 7 |
]
|
| 8 |
+
def list_files(model_name):
|
| 9 |
+
files = api.list_repo_files(repo_id=model_name,repo_type="model")
|
| 10 |
+
return gr.update(choices=[m for m in models])
|
|
|
|
|
|
|
| 11 |
|
| 12 |
def load_models(model_in):
|
| 13 |
loaded_model=[]
|
|
|
|
| 34 |
with gr.Row():
|
| 35 |
model_name=gr.Textbox(label="Model", value=models[0], placeholder=models[0])
|
| 36 |
load_btn=gr.Button("Load")
|
| 37 |
+
with gr.Row():
|
| 38 |
+
file_name=gr.Dropdown(label="Files", choices=[])
|
| 39 |
with gr.Row():
|
| 40 |
models_out=gr.JSON(label="Gradio Details")
|
| 41 |
details=gr.JSON(label="Hub Details")
|
| 42 |
+
app.load(list_files,model_name,[file_name])
|
| 43 |
#app.load(load_models,model_name,[models_out,details])
|
| 44 |
load_btn.click(load_models,model_name,[models_out,details])
|
| 45 |
app.launch()
|