JuanLozada97 commited on
Commit
57ec1a2
1 Parent(s): c90b2b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, 600, 700, 875]])
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