Update app.py

app.py

@@ -47,7 +47,7 @@ def segment_everything(
 
     
     annotations = results[0].masks.data
-    fig = fast_process(annotations=annotations,
+    segmented_img = fast_process(annotations=annotations,
                        image=input,
                        device=device,
                        scale=(1024 // input_size),
@@ -57,8 +57,6 @@ def segment_everything(
                        use_retina=use_retina,
                        withContours=withContours,)
 
-    print('Figure',fig)
-    print('-----------')
     bboxes = results[0].boxes.data
     areas = (bboxes[:, 2] - bboxes[:, 0]) * (bboxes[:, 3] - bboxes[:, 1])
     _, largest_indices = torch.topk(areas, 2)
@@ -82,10 +80,8 @@ def segment_everything(
     
     cropped_img = Image.open(buf).convert("RGBA")
     cropped_img = cropped_img.resize((1024, 682))
-    print('Crop type',cropped_img)
-    print('-----------')
     
-    return fig
+    return segmented_img, cropped_img
 
 title = "<center><strong><font size='8'>🏃 Fast Segment Anything 🤗</font></strong></center>"
 description = """ # 🎯 Instructions for points mode """
@@ -106,6 +102,7 @@ demo = gr.Interface(
     ],
     outputs = [
         gr.Image(label="Segmented Image", interactive=False, type='pil')
+        gr.Image(label="Cropped Image", interactive=False, type='pil')
     ],
     title = title,
     description = description,