Update app.py

app.py

@@ -45,48 +45,17 @@ models = [
     'Jonjew/NSFWMaster',
 ]
 
-# Загрузка моделей
-def load_fn(models):
-    global models_load
-    models_load = {}
-    for model in models:
-        try:
-            models_load[model] = gr.load(f'models/{model}')
-        except Exception as e:
-            print(f"Failed to load model {model}: {e}")
-            models_load[model] = gr.Interface(lambda txt: None, ['text'], ['image'])
-
-load_fn(models)
-
-num_models = len(models)
-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 gen_fn(model_str, prompt, width, height):
-    if model_str == 'NA' or model_str not in models_load:
-        return None
-    try:
-        result = models_load[model_str](prompt)
-        if isinstance(result, str):
-            result = Image.open(result)
-        if isinstance(result, Image.Image):
-            result = result.resize((width, height))
-        return result
-    except Exception as e:
-        print(f"Error generating image for model {model_str}: {e}")
-        noise = str(randint(0, 9999999))
-    return models_load[model_str](f'{prompt} {noise}')
-
-        
+  
                     
+max_images = 8
+MAX_SEED = 2**32-1
+load_models(models)
+
+css = """
+.model_info { text-align: center; }
+.output { width=112px; height=112px; max_width=112px; max_height=112px; !important; }
+.gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
+"""
 
 
 # Функция основного интерфейса