Update app.py

app.py

@@ -218,7 +218,7 @@ def model_process(input):
     print(f"process time_2_: {(time.time() - start) * 1000}ms, {alpha_channel.shape}, {res_np_img.shape} / {res_np_img[250][250]} / {res_np_img.dtype}")
     image = Image.fromarray(res_np_img.astype(np.uint8))
     image.save(f'./result_image.png')
-    return res_np_img.astype(np.uint8) # image
+    return image_pil # res_np_img.astype(np.uint8) # image
     
     '''
     ext = get_image_ext(origin_image_bytes)
@@ -480,7 +480,7 @@ with image_blocks as demo:
                 
 
             # btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
-            btn.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
+            btn.click(fn=predict, inputs=[image], outputs=[image_out]) #, community_icon, loading_icon, share_button])
             #share_button.click(None, [], [], _js=share_js)