remotewith commited on
Commit
bf7e805
·
1 Parent(s): 57474b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def predict(img) -> Tuple[Dict, float]:
52
  img1 = effnetb2_transforms(img1).unsqueeze(0)
53
  pix = normalize_2d(np.array(img))
54
 
55
- pix1=model.predict(img, confidence=40, overlap=30)
56
 
57
 
58
  # Put model into evaluation mode and turn on inference mode
 
52
  img1 = effnetb2_transforms(img1).unsqueeze(0)
53
  pix = normalize_2d(np.array(img))
54
 
55
+ pix1=model.predict(str(img), confidence=40, overlap=30)
56
 
57
 
58
  # Put model into evaluation mode and turn on inference mode