cleanup

app.py

@@ -18,8 +18,6 @@ def infer(img):
   image = img["background"].convert("RGB")
   point_prompt = img["layers"][0]
   total_image = img["composite"]
-  #torch_img = torch.from_numpy(np.array(image))
-  #torch_img = torch_img.permute(2, 0, 1)
   predictor.set_image(np.array(image))
 
   # get point prompt
@@ -51,6 +49,6 @@ with gr.Blocks() as demo:
             )
             submit_btn = gr.Button()
         output = gr.AnnotatedImage()
-    submit_btn.click(infer, inputs=im, outputs=gr.AnnotatedImage())
+    submit_btn.click(infer, inputs=im, outputs=output)
 
 demo.launch(debug=True)