a

app.py

@@ -1,18 +1,20 @@
 import gradio as gr
 import os.path
+from PIL import Image
 
 def output(image_path):
     output_path = 'output'
     caption = 'person . mask'
     flag = False
     save_image_path = os.path.join(output_path,os.path.split(image_path)[-1])
-    return save_image_path
+    save_image = Image.open(save_image_path).convert("RGB")
+    return save_image
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Demo</center></h1>")
     img_obj = gr.Image(value ='image/test.jpg',type = "filepath",label='object_img')
     btn = gr.Button("Submit",variant="primary")
-    img_output = gr.Image(type = "filepath",label='output_img')
+    img_output = gr.Image(type = "pil",label='output_img')
     btn.click(fn=output,inputs=[img_obj],
     outputs=[img_output],
     )

requirements.txt

@@ -0,0 +1 @@
+pillow