Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import numpy
+from PIL import Image
 
 rng = range(1,225)
 sobel_verticle = torch.tensor([[-1,0,1],
@@ -34,7 +35,7 @@ def convolve(img, kernel):
     ch2 = torch.tensor([[apply_kernel(ch2, i,j,kernel) for j in rng] for i in rng])
     ch3 = torch.tensor([[apply_kernel(ch3, i,j,kernel) for j in rng] for i in rng])
     out_img = (torch.stack((ch1,ch2,ch3),-1)).numpy()
-    return out_img
+    return Image.fromarray(out_img)
 
 
 def app(img, filter):