fixed iamge formatting bug

app.py

@@ -9,6 +9,7 @@ device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cp
 
 model = Autoencoder()
 model.load_state_dict(torch.load('model.pt', map_location=device))
+model = model.eval()
 
 resize = Resize((224))
 to_tensor = ToTensor()
@@ -18,7 +19,9 @@ transforms = [to_tensor, resize]
 def test(image):
     for transform in transforms:
         image = transform(image)
-    return [image]
+    image = image.unsqueeze(0)
+    image = model(image).squeeze(0).permute(1,2,0).cpu().detach().numpy()
+    return image
 
 
 interface = gr.Interface(
@@ -26,15 +29,11 @@ interface = gr.Interface(
     description = "Select a image",
     allow_flagging="never",
     fn = test, 
-    inputs = [
-        gr.Image(label = "x", shape = [224, 224]),
-    ],
-    outputs = [
-        gr.Image(label = "pred"),
-    ],
+    inputs = gr.Image(label = "x", type='numpy'),
+    outputs = gr.Image(label = "pred"),
     examples = [
         ["img.jpg"],
     ]
 )
 
-interface.launch(share = False)
+interface.launch()