Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import numpy as np
 
 
+import torch
 from super_image import EdsrModel, ImageLoader
 from PIL import Image
 import requests
@@ -19,8 +20,13 @@ def transformation(image):
     model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=4)
     inputs = ImageLoader.load_image(image)
     preds = model(inputs)
-    preds=np.array(preds)
-    print(preds)
+    print("1  :",preds)
+    print( type(preds) )
+    preds=preds.numpy()
+    
+
+    #preds=np.array(preds)
+    print("2 :",preds)
     ImageLoader.save_image(preds, './scaled_2x.png')
     ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png')