Kev09 commited on
Commit
0fc7951
1 Parent(s): b3c88ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ def transformation(image):
35
  model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
36
  inputs = ImageLoader.load_image(image)
37
  preds = model(inputs)
38
- ImageLoader.save_image(preds, './scaled_2x.png')
39
- ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')
40
  prednumpy=preds.detach().numpy()
41
  return prednumpy
42
 
 
35
  model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
36
  inputs = ImageLoader.load_image(image)
37
  preds = model(inputs)
38
+ # ImageLoader.save_image(preds, './scaled_2x.png')
39
+ # ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')
40
  prednumpy=preds.detach().numpy()
41
  return prednumpy
42