Update app.py

app.py

@@ -6,7 +6,8 @@ import torch
 from super_image import EdsrModel, ImageLoader
 from PIL import Image
 import requests
-
+import torchvision
+import torchvision.transforms as T
 
 
 def greet(name):
@@ -40,11 +41,12 @@ def transformation(image):
   #  ImageLoader.save_image(preds, './scaled_2x.png')
     ImageLoader.save_compare(inputs, preds, 'scaleed_2x_compare.png')
     print("ok2")
-    prednumpy=preds.detach().numpy()
+    #prednumpy=preds.detach().numpy()
+    prednumpy=T.ToPILImage(preds)
     print('pnump',type(prednumpy))
     print('predtype',type(preds))
     print('ok3')
-    prednumpy = np.squeeze(prednumpy)
+   # prednumpy = np.squeeze(prednumpy)
 
     return prednumpy
 
@@ -59,7 +61,7 @@ with gr.Blocks() as demo:
     image1=gr.Image(type='filepath')
     
     button=gr.Button("LE BOUTON")
-    image2=gr.Image(type='numpy')
+    image2=gr.Image(type='pil')
     button.click(fn=transformation,inputs=image1,outputs=image2,api_name="upscale")