Update app.py

app.py

@@ -123,7 +123,9 @@ MAX_IMAGE_SIZE = 4096
 @spaces.GPU(duration=90)
 def infer(
     prompt,
-    negative_prompt,
+    negative_prompt_1,
+    negative_prompt_2,
+    negative_prompt_3,
     seed,
     randomize_seed,
     width,
@@ -191,6 +193,7 @@ def infer(
     else:
         enhanced_prompt = prompt
         enhanced_prompt_2 = prompt
+    model.to('cpu')
     if latent_file:  # Check if a latent file is provided
       #  initial_latents = pipe.prepare_latents(
       #      batch_size=1,
@@ -207,7 +210,7 @@ def infer(
         #with torch.no_grad():
         sd_image = pipe(
             prompt=enhanced_prompt,  # This conversion is fine
-            negative_prompt=negative_prompt,
+            negative_prompt=negative_prompt_1,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
             width=width,
@@ -225,7 +228,9 @@ def infer(
             prompt=prompt,  # This conversion is fine
             prompt_2=enhanced_prompt_2,
             prompt_3=enhanced_prompt,
-            negative_prompt=negative_prompt,
+            negative_prompt=negative_prompt_1,
+            negative_prompt_2=negative_prompt_2,
+            negative_prompt_3=negative_prompt_3,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
             width=width,
@@ -235,7 +240,6 @@ def infer(
             generator=generator,
             max_sequence_length=512
         ).images[0]
-        
         print('-- got image --')
         #sd35_image = pipe.vae.decode(sd_image / 0.18215).sample
        # sd35_image = sd35_image.cpu().permute(0, 2, 3, 1).float().detach().numpy()
@@ -244,7 +248,6 @@ def infer(
       #  sd35_path = f"sd35_{seed}.png"
        # image_pil.save(sd35_path,optimize=False,compress_level=0)
        # upload_to_ftp(sd35_path)
-
         sd35_path = f"sd35_{seed}.png"
         sd_image.save(sd35_path,optimize=False,compress_level=0)
         upload_to_ftp(sd35_path)
@@ -355,11 +358,26 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
         result = gr.Image(label="Result", show_label=False)
         with gr.Accordion("Advanced Settings", open=True):
             latent_file = gr.File(label="Image File (optional)")  # Add latents file input
-            negative_prompt = gr.Text(
-                label="Negative prompt",
+            negative_prompt_1 = gr.Text(
+                label="Negative prompt 1",
                 max_lines=1,
                 placeholder="Enter a negative prompt",
                 visible=True,
+                value="bad anatomy, poorly drawn hands, distorted face, blurry, out of frame, low resolution, grainy, pixelated, disfigured, mutated, extra limbs, bad composition"
+            )
+            negative_prompt_2 = gr.Text(
+                label="Negative prompt 2",
+                max_lines=1,
+                placeholder="Enter a second negative prompt",
+                visible=True,
+                value="unrealistic, cartoon, anime, sketch, painting, drawing, illustration, graphic, digital art, render, 3d, blurry, deformed, disfigured, poorly drawn, bad anatomy, mutated, extra limbs, ugly, out of frame, bad composition, low resolution, grainy, pixelated, noisy, oversaturated, undersaturated, (worst quality, low quality:1.3), (bad hands, missing fingers:1.2)"
+            )
+            negative_prompt_3 = gr.Text(
+                label="Negative prompt 3",
+                max_lines=1,
+                placeholder="Enter a third negative prompt",
+                visible=True,
+                value="(worst quality, low quality:1.3), (bad anatomy, bad hands, missing fingers, extra digit, fewer digits:1.2), (blurry:1.1), cropped, watermark, text, signature, logo, jpeg artifacts, (ugly, deformed, disfigured:1.2), (poorly drawn:1.2), mutated, extra limbs, (bad proportions, gross proportions:1.2), (malformed limbs, missing arms, missing legs, extra arms, extra legs:1.2), (fused fingers, too many fingers, long neck:1.2), (unnatural body, unnatural pose:1.1), out of frame, (bad composition, poorly composed:1.1), (oversaturated, undersaturated:1.1), (grainy, pixelated:1.1), (low resolution, noisy:1.1), (unrealistic, distorted:1.1), (extra fingers, mutated hands, poorly drawn hands, bad hands:1.3), (missing fingers:1.3)"
             )
             num_iterations = gr.Number(
                 value=1000, 
@@ -407,7 +425,9 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
         fn=infer,
         inputs=[
             prompt,
-            negative_prompt,
+            negative_prompt_1,
+            negative_prompt_2,
+            negative_prompt_3,
             seed,
             randomize_seed,
             width,