Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 from pyChatGPT import ChatGPT
-import cloudpickle
+#import cloudpickle
 
 os.system("pip install -U gradio")
 
@@ -98,9 +98,9 @@ def inference(img,unique_only):
     object_list_str = []
 
     
-    masks = outputs["instances"].pred_masks
-    masks_screen = masks.cpu().numpy()
-    print(masks_screen)
+    #masks = outputs["instances"].pred_masks
+    #masks_screen = masks.cpu().numpy()
+    #print(masks_screen)
 
     box_locations = outputs["instances"].pred_boxes
     box_loc_screen = box_locations.tensor.cpu().numpy()
@@ -132,8 +132,8 @@ def inference(img,unique_only):
 
     return (
         #Image.fromarray(np.uint8(out.get_image())).convert("RGB"),
-        json.dumps(detected_objects),
-        cloudpickle.dumps(masks_screen)
+        json.dumps(detected_objects)
+        #cloudpickle.dumps(masks_screen)
     )
 
 
@@ -145,8 +145,8 @@ with gr.Blocks() as demo:
     with gr.Column():
         #outviz = gr.Image(label="Visualization", type="pil")
         output_desc = gr.JSON(label="Detected Objects")
-        masks_screen = gr.Textbox(label="Masks binary", lines=1)
+        #masks_screen = gr.Textbox(label="Masks binary", lines=1)
 
-    btn_detic.click(fn=inference, inputs=[inp,chk], outputs=[output_desc,masks_screen], api_name="detect")
+    btn_detic.click(fn=inference, inputs=[inp,chk], outputs=[output_desc], api_name="detect")
 
 demo.launch()