Omnibus commited on
Commit
68b46bf
·
verified ·
1 Parent(s): 09d1825

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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=models[0],repo_type="model")
10
- print(files)
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()