Update app.py

app.py

@@ -213,10 +213,11 @@ def read_content(file_path: str) -> str:
 
     return content
 
-def predict(dict, prompt=""):
+def predict(dict):
     print(f'liuyz_0_', dict)
     init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
-    print(f'liuyz_1_', init_image)
+    target_size = (init_image.shape[0], init_image.shape[1])
+    print(f'liuyz_1_', init_image, target_size)
     print(f'liuyz_2_', init_image.convert("RGB"))
     print(f'liuyz_3_', init_image.convert("RGB").resize((512, 512)))
     mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
@@ -224,7 +225,7 @@ def predict(dict, prompt=""):
     output = None # output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     
-    return output.images[0], gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 print(f'liuyz_400_here_')
 
@@ -279,8 +280,8 @@ interface.launch(share=True)
 
 image = gr.inputs.Image(source='upload', tool='sketch', type="pil", label="Upload")# .style(height=400)
 demo = gr.Interface(fn=predict, 
-             inputs=image,
-             outputs=image,
+             inputs=[image],
+             outputs=[image],
              # examples=[["cheetah.jpg"]],
              )