TheXeos commited on
Commit
30e7031
·
1 Parent(s): 458c769

Cleanup and select default model

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -1,10 +1,4 @@
1
- import time
2
  import gradio as gr
3
- import os
4
-
5
- import numpy as np
6
- from PIL import Image
7
- import math
8
  from collections import defaultdict
9
  import os
10
  from huggingface_hub import HfApi, hf_hub_download
@@ -58,7 +52,7 @@ def update_models(class_name):
58
 
59
  def update_model_list(class_name):
60
  model_choices = dataset_dict[class_name]
61
- return gr.Dropdown.update(choices=model_choices)
62
 
63
 
64
  with gr.Blocks() as demo:
 
 
1
  import gradio as gr
 
 
 
 
 
2
  from collections import defaultdict
3
  import os
4
  from huggingface_hub import HfApi, hf_hub_download
 
52
 
53
  def update_model_list(class_name):
54
  model_choices = dataset_dict[class_name]
55
+ return gr.Dropdown.update(choices=model_choices, value=model_choices[0])
56
 
57
 
58
  with gr.Blocks() as demo: