remotewith commited on
Commit
8b197dc
·
1 Parent(s): 6fc1fce

Update app.py

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