Lewislou commited on
Commit
2f421e1
·
1 Parent(s): 73b79a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def normalize_channel(img, lower=1, upper=99):
19
  img_norm = img
20
  return img_norm.astype(np.uint8)
21
 
22
- def predict(filename, model=None, device=None, reduce_labels=True):
23
  if img_name.endswith('.tif') or img_name.endswith('.tiff'):
24
  img_data = tif.imread(img_name)
25
  else:
 
19
  img_norm = img
20
  return img_norm.astype(np.uint8)
21
 
22
+ def predict(img_name, model=None, device=None, reduce_labels=True):
23
  if img_name.endswith('.tif') or img_name.endswith('.tiff'):
24
  img_data = tif.imread(img_name)
25
  else: