Update app.py

app.py

@@ -59,12 +59,15 @@ def load_fn(models):
 load_fn(models)
 
 num_models = len(models)
-default_models = models[:4]  # Выбираем первые 4 модели по умолчанию
+default_models = models[:num_models]
 
-# Расширение списка для выбора
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']
 
+def update_imgbox(choices):
+    choices_plus = extend_choices(choices)
+    return [gr.Image(None, label=m, visible=(m != 'NA'), elem_id="custom_image") for m in choices_plus]
+
 # Обновление изображения
 def update_imgbox(choices, width, height):
     choices_plus = extend_choices(choices)