Update app.py

app.py

@@ -58,18 +58,17 @@ def segment_everything(
                        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)
     largest_boxes = bboxes[largest_indices]
     for i, box in enumerate(largest_boxes):
         print(f"Largest Box {i+1}: {box.tolist()}")
-
-    print('Type',input)
+    print('-----------')
+    
     fig, ax = plt.subplots(1)
     ax.imshow(input)
-
     for box in largest_boxes:
         x1, y1, x2, y2 = box[:4] 
         rect = patches.Rectangle((x1, y1), x2-x1, y2-y1, linewidth=2, edgecolor='r', facecolor='none')
@@ -84,8 +83,9 @@ 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,cropped_img
+    return fig
 
 title = "<center><strong><font size='8'>🏃 Fast Segment Anything 🤗</font></strong></center>"
 description = """ # 🎯 Instructions for points mode """
@@ -93,9 +93,6 @@ examples = [["examples/sa_8776.jpg"], ["examples/sa_414.jpg"], ["examples/sa_130
             ["examples/sa_561.jpg"], ["examples/sa_192.jpg"], ["examples/sa_10039.jpg"], ["examples/sa_862.jpg"]]
 default_example = examples[0]
 
-cond_img_e = gr.Image(label="Input", value=default_example[0], type='pil')
-segm_img_e = gr.Image(label="Segmented Image", interactive=False, type='pil')
-
 input_size_slider = gr.components.Slider(minimum=512,maximum=1024,value=1024,step=64,label='Input_size',info='Our model was trained on a size of 1024')
 
 css = "h1 { text-align: center } .about { text-align: justify; padding-left: 10%; padding-right: 10%; }"
@@ -108,8 +105,7 @@ demo = gr.Interface(
         gr.Checkbox(value=True, label='withContours', info='draw the edges of the masks')
     ],
     outputs = [
-        gr.Image(label="Segmented Image", interactive=False, type='pil'),
-        gr.Image(label="Cropped Image", interactive=False, type='pil')
+        gr.Image(label="Segmented Image", interactive=False, type='pil')
     ],
     title = title,
     description = description,