Update app.py

app.py

@@ -50,7 +50,6 @@ def segment_to_bbox(segment_indexs):
         return [0.0,0.0,0.0,0.0]
 
 def clipseg_prediction(image):
-
     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")
@@ -94,8 +93,8 @@ def foward_pass(image_input: np.ndarray, points: List[List[int]]) -> np.ndarray:
         cache_data = [pixels, embedding]
     del inputs["pixel_values"]
 
-    # outputs = model.forward(image_embeddings=cache_data[1], **inputs)
-    outputs = model(**inputs)
+    outputs = model.forward(image_embeddings=cache_data[1], **inputs)
+    # outputs = model(**inputs)
     masks = processor.image_processor.post_process_masks(
         outputs.pred_masks.cpu(), inputs["original_sizes"].to(device), inputs["reshaped_input_sizes"].to(device)
     )