Update app.py

app.py

@@ -9,6 +9,9 @@ import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 import io
 
+import warnings
+warnings.filterwarnings(action='ignore')
+
 model = YOLO('./weights/FastSAM-x.pt')
 
 device = torch.device(
@@ -60,7 +63,8 @@ def segment_everything(
     largest_boxes = bboxes[largest_indices]
     for i, box in enumerate(largest_boxes):
         print(f"Largest Box {i+1}: {box.tolist()}")
-    
+
+    print('Type',input)
     fig, ax = plt.subplots(1)
     ax.imshow(input)
 
@@ -68,7 +72,8 @@ def segment_everything(
         x1, y1, x2, y2 = box[:4] 
         rect = patches.Rectangle((x1, y1), x2-x1, y2-y1, linewidth=2, edgecolor='r', facecolor='none')
         ax.add_patch(rect)
-    
+
+    print('Done till here')
     ax.axis('off')
     buf = io.BytesIO()
     plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)