Add application file

app.py

@@ -22,6 +22,7 @@ def pil_loader(filepath):
 
 def build_transforms(input_size, center_crop=True):
     transform = torchvision.transforms.Compose([
+        torchvision.transforms.ToPILImage()
         torchvision.transforms.Resize(input_size * 8 // 7),
         torchvision.transforms.CenterCrop(input_size),
         torchvision.transforms.ToTensor(),