Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 import torch
 import modin.pandas as pd
 from diffusers import DiffusionPipeline 
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 if torch.cuda.is_available():
-    PYTORCH_CUDA_ALLOC_CONF = {'max_split_size_mb': 6000}
+    PYTORCH_CUDA_ALLOC_CONF={'max_split_size_mb': 6000}
     torch.cuda.max_memory_allocated(device=device)
     torch.cuda.empty_cache()
     pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
@@ -19,35 +20,21 @@ else:
     pipe = pipe.to(device)
     refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", use_safetensors=True)
     refiner = refiner.to(device)
-def genie(prompt, negative_prompt, height, width, scale, steps, seed, strength):
+
+def genie (prompt, negative_prompt, height, width, scale, steps, seed, strength):
     generator = torch.Generator(device=device).manual_seed(seed)
     int_image = pipe(prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator, output_type="latent").images 
     image = refiner(prompt=prompt, negative_prompt=negative_prompt, image=int_image).images[0]
     return image
-models = [
-    {"name": "Модель 1", "value": "models/cag/anything-v3-1"},
-    {"name": "Модель 2", "value": "models/emilianJR/AnyLORA"}
-]
-current_model = models[0]
-def set_model(selected_model):
-    global current_model
-    current_model = models[selected_model]
-input_prompt = gr.inputs.Textbox(label='Что вы хотите, чтобы ИИ генерировал')
-input_negative_prompt = gr.inputs.Textbox(label='Что вы не хотите, чтобы ИИ генерировал')
-input_height = gr.inputs.Slider(512, 1024, 768, step=128, label='Высота')
-input_width = gr.inputs.Slider(512, 1024, 768, step=128, label='Ширина')
-input_scale = gr.inputs.Slider(1, 15, 10, label='Шкала навигации')
-input_steps = gr.inputs.Slider(25, maximum=50, value=25, step=1, label='Количество итераций')
-input_seed = gr.inputs.Slider(label="Зерно", minimum=0, maximum=987654321987654321, step=1, randomize=True)
-input_strength = gr.inputs.Slider(label='Сила', minimum=0, maximum=1, step=0.05, default_value=0.5)
-model_names = [m["name"] for m in models]
-model_selection = gr.inputs.Dropdown(
-    choices=model_names,
-    type="index",
-    label="Выберите модель",
-    default=0,
-    onchange=set_model
-)
-output_image = gr.outputs.Image(label="Результат")
-iface = gr.Interface(fn=genie, inputs=[input_prompt, input_negative_prompt, input_height, input_width, input_scale, input_steps, input_seed, input_strength, model_selection], outputs=output_image, title="Стабильная Диффузия - SDXL - txt2img", article="")
-iface.launch()
+    
+gr.Interface(fn=genie, inputs=[gr.Textbox(label='Что вы хотите, чтобы ИИ генерировал'), 
+    gr.Textbox(label='Что вы не хотите, чтобы ИИ генерировал'), 
+    gr.Slider(512, 1024, 768, step=128, label='Высота'),
+    gr.Slider(512, 1024, 768, step=128, label='Ширина'),
+    gr.Slider(1, 15, 10, label='Шкала навигации'), 
+    gr.Slider(25, maximum=50, value=25, step=1, label='Количество итераций'), 
+    gr.Slider(label="Зерно", minimum=0, maximum=987654321987654321, step=1, randomize=True), 
+        gr.Slider(label='Сила', minimum=0, maximum=1, step=.05, value=.5)], 
+    outputs='image', 
+    title="Стабильная Диффузия - SDXL - txt2img", 
+        article = "<br><br><br><br><br>").launch(debug=True, max_threads=80)