piyushgrover commited on
Commit
3906e17
·
1 Parent(s): 8200379

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -202,7 +202,7 @@ with gr.Blocks() as app:
202
  x = plt.show()
203
 
204
 
205
- def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.6, show_cam: bool = False,
206
  transparency: float = 0.5) -> List[np.ndarray]:
207
  with torch.no_grad():
208
  transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)
 
202
  x = plt.show()
203
 
204
 
205
+ def predict(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.5, show_cam: bool = False,
206
  transparency: float = 0.5) -> List[np.ndarray]:
207
  with torch.no_grad():
208
  transformed_image = cfg.grad_cam_transforms(image=image)["image"].unsqueeze(0)