Update app.py

app.py

@@ -182,13 +182,14 @@ def b64_to_pil_list(b64_json_str):
 
 def update_dimensions_on_upload(image):
     if image is None:
-        return 1024, 1024
+        return 1536, 1536  # Увеличили базовый размер
     w, h = image.size
+    target_dim = 1536  # Можно поставить 2048 для еще более высокого разрешения
     if w > h:
-        nw = 1024
+        nw = target_dim
         nh = int(nw * h / w)
     else:
-        nh = 1024
+        nh = target_dim
         nw = int(nh * w / h)
     return (nw // 8) * 8, (nh // 8) * 8
 
@@ -205,7 +206,7 @@ def infer(images_b64_json, prompt, seed, randomize_seed, guidance_scale, steps,
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
-    negative_prompt = ""
+    negative_prompt = "blurry, low resolution, low quality, soft details, artifacts, pixelated"
     width, height = update_dimensions_on_upload(pil_images[0])
     try:
         result_image = pipe(
@@ -984,8 +985,8 @@ with gr.Blocks() as demo:
     prompt = gr.Textbox(value="", elem_id="prompt-gradio-input", elem_classes="hidden-input", container=False)
     seed = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, value=0, elem_id="gradio-seed", elem_classes="hidden-input", container=False)
     randomize_seed = gr.Checkbox(value=True, elem_id="gradio-randomize", elem_classes="hidden-input", container=False)
-    guidance_scale = gr.Slider(minimum=1.0, maximum=10.0, step=0.1, value=1.0, elem_id="gradio-guidance", elem_classes="hidden-input", container=False)
-    steps = gr.Slider(minimum=1, maximum=50, step=1, value=4, elem_id="gradio-steps", elem_classes="hidden-input", container=False)
+    guidance_scale = gr.Slider(minimum=1.0, maximum=10.0, step=0.1, value=3.5, elem_id="gradio-guidance", elem_classes="hidden-input", container=False)
+    steps = gr.Slider(minimum=1, maximum=50, step=1, value=20, elem_id="gradio-steps", elem_classes="hidden-input", container=False)
     result = gr.Image(elem_id="gradio-result", elem_classes="hidden-input", container=False, format="png")
 
     example_idx = gr.Textbox(value="", elem_id="example-idx-input", elem_classes="hidden-input", container=False)