Update app.py

app.py

@@ -119,9 +119,10 @@ def generate(prompt, steps, num_iterations, feel, seed):
 
     wav_bytes, duration_s = wav_bytes_from_spectrogram_image(stitched_image)
 
-    mask = Image.new("RGB", (512, 512), color="white")
-    bg_image = outpaint(prompt, init_image, mask, steps)
-    bg_image.save("bg_image.png")
+    # mask = Image.new("RGB", (512, 512), color="white")
+    # bg_image = outpaint(prompt, init_image, mask, steps)
+    # bg_image.save("bg_image.png")
+    init_image.save("bg_image.png")
 
     # return read(wav_bytes)
     with open("output.wav", "wb") as f:
@@ -191,9 +192,10 @@ def generate_riffuse(prompt_start, steps, num_iterations, feel, prompt_end=None,
     # return read(wav_bytes)
     # return wav_list_to_wav(wav_list)
 
-    mask = Image.new("RGB", (512, 512), color="white")
-    bg_image = outpaint(f"{prompt_start} and {prompt_end}", init_image, mask, steps)
-    bg_image.save("bg_image.png")
+    # mask = Image.new("RGB", (512, 512), color="white")
+    # bg_image = outpaint(f"{prompt_start} and {prompt_end}", init_image, mask, steps)
+    # bg_image.save("bg_image.png")
+    init_image.save("bg_image.png")
 
     with open("output.wav", "wb") as f:
         f.write(wav_list_to_wav(wav_list))