Update app.py

app.py

@@ -69,8 +69,6 @@ async def generate_image(
         num_inference_steps=steps,
         model=basemodel,
     )
-    image1 = Image.open(image1)
-    print(image1)
     image2 = await client.text_to_image(
         prompt=text,
         height=height,
@@ -79,9 +77,7 @@ async def generate_image(
         num_inference_steps=steps,
         model=model,
     )
-    print(image2)
-    image2 = Image.open(image2)
-    return image1, image2, seed
+    yield image1, image2, seed
 
 async def gen(
     prompt:str,
@@ -96,7 +92,7 @@ async def gen(
 ):
     model = enable_lora(lora_in, lora_add)
     image1, image2, seed = await generate_image(prompt,model,width,height,scales,steps,seed)
-    return image1, image2, seed
+    yield image1, image2, seed
      
 
 
@@ -133,8 +129,8 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
     with gr.Row():
         with gr.Column(scale=4):
             with gr.Row():
-                img1 = gr.Gallery(label='flux Generated Image', columns = 1, preview=True, height=600)
-                img2 = gr.Gallery(label='LoRA Generated Image', columns = 1, preview=True, height=600)        
+                img1 = gr.Image(type="PIL", label='flux Generated Image', height=600)
+                img2 = gr.Image(type="PIL", label='LoRA Generated Image', height=600)        
             with gr.Row():
                 prompt = gr.Textbox(label='Enter Your Prompt (Multi-Languages)', placeholder="Enter prompt...", scale=6)
                 sendBtn = gr.Button(scale=1, variant='primary')