Update app.py

app.py

@@ -51,7 +51,7 @@ def generate_image(images, prompt, negative_prompt,instruct_query, scale, nfaa_n
     faceid_all_embeds = []
     first_iteration = True
     image = images
-
+    yield None
     total_negative_prompt = f"{negative_prompt} {nfaa_negative_prompt}"
     print("Generating normal")
 
@@ -83,7 +83,8 @@ def generate_image(images, prompt, negative_prompt,instruct_query, scale, nfaa_n
         num_inference_steps=30,
         query=instruct_query
     )
-    return image
+    
+    yield image