JuanLozada97
commited on
Commit
•
57ec1a2
1
Parent(s):
c90b2b6
Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ def predict(img) -> Tuple[Dict, float]:
|
|
104 |
with torch.inference_mode():
|
105 |
image_embedding = medsam_model.image_encoder(img_1024_tensor) # (1, 256, 64, 64)
|
106 |
# define the inputbox
|
107 |
-
input_box = np.array([[425,
|
108 |
# transfer box_np t0 1024x1024 scale
|
109 |
box_1024 = input_box / np.array([W, H, W, H]) * 1024
|
110 |
|
|
|
104 |
with torch.inference_mode():
|
105 |
image_embedding = medsam_model.image_encoder(img_1024_tensor) # (1, 256, 64, 64)
|
106 |
# define the inputbox
|
107 |
+
input_box = np.array([[425, 200, 600, 400]])
|
108 |
# transfer box_np t0 1024x1024 scale
|
109 |
box_1024 = input_box / np.array([W, H, W, H]) * 1024
|
110 |
|