Update app.py

app.py

@@ -77,7 +77,7 @@ async def generate_image(
         num_inference_steps=steps,
         model=model,
     )
-    yield image1, image2, seed
+    return image1, image2, seed
 
 async def gen(
     prompt:str,
@@ -91,8 +91,8 @@ async def gen(
     progress=gr.Progress(track_tqdm=True)
 ):
     model = enable_lora(lora_in, lora_add)
-    image1, image2, seed = generate_image(prompt,model,width,height,scales,steps,seed)
-    yield image1, image2, seed
+    image1, image2, seed = await generate_image(prompt,model,width,height,scales,steps,seed)
+    return image1, image2, seed