Update app.py

app.py

@@ -775,21 +775,6 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-def pipe_generate_image(p1,p2,h,w):
-    for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
-            prompt=p1,
-            negative_prompt=p2,
-            height=h,
-            width=w,
-            output_type="pil",
-            guidance_scale=img_accu,
-            num_images_per_prompt=1,
-            num_inference_steps=image_steps,
-            max_sequence_length=seq,
-            generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
-    ):
-        yield img
-
 def add_song_cover_text(img,artist,song,h,w):
 
     draw = ImageDraw.Draw(img,mode="RGBA")
@@ -1458,7 +1443,19 @@ def handle_generation(h,w,d):
         Negative: {neg}
     """)
 
-    yield pipe_generate_image(pos,neg,h,w)
+    for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
+            prompt=pos,
+            negative_prompt=neg,
+            height=h,
+            width=w,
+            output_type="pil",
+            guidance_scale=img_accu,
+            num_images_per_prompt=1,
+            num_inference_steps=image_steps,
+            max_sequence_length=seq,
+            generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
+    ):
+        yield img
     
 # entry