Meloo commited on
Commit
33d897a
1 Parent(s): 6e71a73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -108,7 +108,7 @@ def inference(image, upscale, large_input_flag, color_fix):
108
  if upscale is None or not isinstance(upscale, (int, float)) or upscale == 3.:
109
  upscale = 2
110
 
111
- upscale = int(upscale) # convert type to int
112
 
113
  model = set_safmn(upscale)
114
 
@@ -116,7 +116,7 @@ def inference(image, upscale, large_input_flag, color_fix):
116
  # print(f'input size: {img.shape}')
117
 
118
  # img2tensor
119
- y = image.astype(np.float32) / 255.
120
  y = torch.from_numpy(np.transpose(y[:, :, [2, 1, 0]], (2, 0, 1))).float()
121
  y = y.unsqueeze(0).to(device)
122
 
 
108
  if upscale is None or not isinstance(upscale, (int, float)) or upscale == 3.:
109
  upscale = 2
110
 
111
+ upscale = int(upscale)
112
 
113
  model = set_safmn(upscale)
114
 
 
116
  # print(f'input size: {img.shape}')
117
 
118
  # img2tensor
119
+ y = np.array(image).astype(np.float32) / 255.
120
  y = torch.from_numpy(np.transpose(y[:, :, [2, 1, 0]], (2, 0, 1))).float()
121
  y = y.unsqueeze(0).to(device)
122