Update app.py

app.py

@@ -155,8 +155,6 @@ def infer_impl(
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)
 
-    yield None, None, None, seed
-
     generator = torch.Generator(device=device).manual_seed(seed)
     if style_to_prompt[style] != "":
         prompt = style_to_prompt[style] + ", " + prompt
@@ -236,7 +234,7 @@ def infer_impl(
     image_w_logo.paste(image)
     image_w_logo.paste(likes_logo_top_img, (image.width - likes_logo_top_img.width,0), likes_logo_top_img)
     image_w_logo.paste(likes_logo_img, (0, image.height))
-    yield image_w_logo, license_str, detected_public_fictional_characters_info, seed
+    return image_w_logo, license_str, detected_public_fictional_characters_info, seed
 
 def infer(
     prompt,
@@ -254,7 +252,8 @@ def infer(
 ):
     if any([a in prompt for a in bad_words]):
         return None, "## <span style='color:orangered'>ERROR: Invalid prompt / 不適切なプロンプト </span>", None, seed    
-    return infer_impl(prompt, style, lang, negative_prompt, seed, randomize_seed, ar, guidance_scale, num_inference_steps, progress)
+    yield None, None, None, seed
+    yield infer_impl(prompt, style, lang, negative_prompt, seed, randomize_seed, ar, guidance_scale, num_inference_steps, progress)
 
 
 # Seafoam theme based on