Update app.py

app.py

@@ -64,10 +64,6 @@ 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)
@@ -81,7 +77,6 @@ def update_imgbox(choices, width, height):
             height=height
         ) for m in choices_plus
     ]
-
 # Функция генерации
 def gen_fn(model_str, prompt, width, height):
     if model_str == 'NA' or model_str not in models_load:
@@ -96,6 +91,11 @@ def gen_fn(model_str, prompt, width, height):
     except Exception as e:
         print(f"Error generating image for model {model_str}: {e}")
         return None
+        noise = str(randint(0, 9999999)
+        return models_load[model_str](f'{prompt} {noise}')
+        
+                    
+
 
 # Функция основного интерфейса
 def make_me():