piyushgrover commited on
Commit
3252151
·
1 Parent(s): 588684f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -211,7 +211,7 @@ with gr.Blocks() as app:
211
  x = plt.show()
212
 
213
 
214
- def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.4, show_cam: bool = False,
215
  transparency: float = 0.5) -> List[np.ndarray]:
216
  with torch.no_grad():
217
  transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
 
211
  x = plt.show()
212
 
213
 
214
+ def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.6, show_cam: bool = False,
215
  transparency: float = 0.5) -> List[np.ndarray]:
216
  with torch.no_grad():
217
  transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)