Update app.py

app.py

@@ -48,8 +48,8 @@ def segment_to_bbox(segment_indexs):
 
 def clipseg_prediction(image):
 
-    print('image_shape:: ', image.shape)
-    img_w, img_h = image.shape
+    print('Clip-Segmentation-started------->')
+    img_w, img_h,_ = image.shape
     inputs = clip_processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
     # predict
     with torch.no_grad():
@@ -81,6 +81,7 @@ def clipseg_prediction(image):
 
 @torch.no_grad()
 def foward_pass(image_input: np.ndarray, points: List[List[int]]) -> np.ndarray:
+    print('SAM-Segmentation-started------->')
     global cache_data
     image_input = Image.fromarray(image_input)
     inputs = processor(image_input, input_points=points, return_tensors="pt").to(device)
@@ -112,10 +113,9 @@ def main_func(inputs):
         mask_colors[final_mask, :] = np.array([[128, 0, 0]])
         return Image.fromarray((mask_colors * 0.6 + image_input * 0.4).astype('uint8'), 'RGB')
     else:
+        print('Prediction:: No vehicle found in the image')
         return Image.fromarray(image_input)
 
-    return pred_masks
-
 def reset_data():
     global cache_data
     cache_data = None