Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
)
|