test

app.py

@@ -10,7 +10,7 @@ def pil_to_torch(image):
     image = image.transpose((2, 0, 1))
     image = torch.tensor(image).float() / 255
     image = image.unsqueeze(0)
-    image = torch.nn.functional.interpolate(image.unsqueeze(0), size=(128, 128*image.shape[3]//image.shape[2]))
+    image = torch.nn.functional.interpolate(image, size=(128, 128*image.shape[3]//image.shape[2]))
     return image