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)
|