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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -33,12 +33,14 @@ def transformation(image):
33
  url = 'https://paperswithcode.com/media/datasets/Set5-0000002728-07a9793f_zA3bDjj.jpg'
34
  image = Image.open(requests.get(url, stream=True).raw)
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
 
43
 
44
 
 
33
  url = 'https://paperswithcode.com/media/datasets/Set5-0000002728-07a9793f_zA3bDjj.jpg'
34
  image = Image.open(requests.get(url, stream=True).raw)
35
  model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
36
+ print('ok')
37
  inputs = ImageLoader.load_image(image)
38
  preds = model(inputs)
39
+ print('ok1')
40
  # ImageLoader.save_image(preds, './scaled_2x.png')
41
  # ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')
42
+ #prednumpy=preds.detach().numpy()
43
+ return pred
44
 
45
 
46