charliebaby2023 commited on
Commit
bb4a6ad
·
verified ·
1 Parent(s): 97e8ac1

Update handlemodelradio.py

Browse files
Files changed (1) hide show
  1. handlemodelradio.py +4 -4
handlemodelradio.py CHANGED
@@ -1,10 +1,10 @@
1
-
2
- def extend_choices(choices,num_models):
3
  return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
4
- def update_imgbox(choices,num_models):
5
  choices_plus = extend_choices(choices[:num_models])
6
  return [gr.Image(None, label=m, visible=(m!='NA')) for m in choices_plus]
7
- def random_choices(num_models):
8
  import random
9
  random.seed()
10
  return random.choices(models, k=num_models)
 
1
+ from config import howManyModelsToUse,num_models
2
+ def extend_choices(choices):
3
  return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
4
+ def update_imgbox(choices):
5
  choices_plus = extend_choices(choices[:num_models])
6
  return [gr.Image(None, label=m, visible=(m!='NA')) for m in choices_plus]
7
+ def random_choices():
8
  import random
9
  random.seed()
10
  return random.choices(models, k=num_models)