Update app.py

app.py

@@ -42,13 +42,15 @@ def transformation(image):
     ImageLoader.save_compare(inputs, preds, 'scaleed_2x_compare.png')
     print("ok2")
     #prednumpy=preds.detach().numpy()
-    prednumpy=T.ToPILImage()(preds)
-    print('pnump',type(prednumpy))
+    preds = preds.data.cpu().numpy()
+    pred = preds[0].transpose((1, 2, 0)) * 255.0
+      #  return Image.fromarray(pred.astype('uint8'), 'RGB')
+  #  print('pnump',type(prednumpy))
     print('predtype',type(preds))
     print('ok3')
    # prednumpy = np.squeeze(prednumpy)
 
-    return prednumpy
+    return Image.fromarray(pred.astype('uint8'), 'RGB')