ghlee94 commited on
Commit
be186ba
1 Parent(s): 587d678

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = io.imread(img_name)
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)