Abijith commited on
Commit
6069b61
·
1 Parent(s): 1114c33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -94,7 +94,8 @@ def foward_pass(image_input: np.ndarray, points: List[List[int]]) -> np.ndarray:
94
  cache_data = [pixels, embedding]
95
  del inputs["pixel_values"]
96
 
97
- outputs = model.forward(image_embeddings=cache_data[1], **inputs)
 
98
  masks = processor.image_processor.post_process_masks(
99
  outputs.pred_masks.cpu(), inputs["original_sizes"].to(device), inputs["reshaped_input_sizes"].to(device)
100
  )
 
94
  cache_data = [pixels, embedding]
95
  del inputs["pixel_values"]
96
 
97
+ # outputs = model.forward(image_embeddings=cache_data[1], **inputs)
98
+ outputs = model(**inputs)
99
  masks = processor.image_processor.post_process_masks(
100
  outputs.pred_masks.cpu(), inputs["original_sizes"].to(device), inputs["reshaped_input_sizes"].to(device)
101
  )