Update handlemodelradio.py

handlemodelradio.py

@@ -1,11 +1,10 @@
-
-
-def extend_choices(choices):
+    
+def extend_choices(choices,num_models):
     return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
-def update_imgbox(choices):
+def update_imgbox(choices,num_models):
     choices_plus = extend_choices(choices[:num_models])
     return [gr.Image(None, label=m, visible=(m!='NA')) for m in choices_plus]
-def random_choices():
+def random_choices(num_models):
     import random
     random.seed()
     return random.choices(models, k=num_models)