Update app.py
Browse files
app.py
CHANGED
@@ -1135,7 +1135,7 @@ def predict(img):
|
|
1135 |
if img_name.endswith('.tif') or img_name.endswith('.tiff'):
|
1136 |
origin_img = tif.imread(img_name)
|
1137 |
else:
|
1138 |
-
origin_img =
|
1139 |
|
1140 |
img_data = pred_transforms(img_name)
|
1141 |
img_data = img_data.to(device)
|
|
|
1135 |
if img_name.endswith('.tif') or img_name.endswith('.tiff'):
|
1136 |
origin_img = tif.imread(img_name)
|
1137 |
else:
|
1138 |
+
origin_img = io_imread(img_name)
|
1139 |
|
1140 |
img_data = pred_transforms(img_name)
|
1141 |
img_data = img_data.to(device)
|