Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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:
|