Update app.py

app.py

@@ -54,7 +54,6 @@ load_fn(models)"""
     
 
 def extend_choices(choices):
-    nb_models
     return choices + (nb_models - len(choices)) * ['NA']
     """return choices + (num_models - len(choices)) * ['NA']"""
 
@@ -75,7 +74,8 @@ def choice_group_a(group_model_choice):
 
 def choice_group_b(group_model_choice):
     choice=choice_group_a(group_model_choice)
-    return [gr.Image(label=m, min_width=170, height=170) for m in choice]
+    """return [gr.Image(label=m, min_width=170, height=170) for m in choice]"""
+    return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choice]
 
 def choice_group_c(group_model_choice):
     choice=choice_group_a(group_model_choice)
@@ -124,8 +124,8 @@ def make_me():
             choices=models_test[0]
             """output = [gr.Image(label=m, min_width=170, height=170) for m in choices]
             current_models = [gr.Textbox(m, visible=False) for m in choices]"""
-            output = update_imgbox([choices[0][0]])
-            current_models = extend_choices_b([choices[0][0]])
+            output = update_imgbox([choices[0]])
+            current_models = extend_choices_b([choices[0]])
             
             for m, o in zip(current_models, output):
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)