Update app.py

app.py

@@ -13,6 +13,15 @@ weights = torch.load('./checkpoints/caption_model.pt', map_location=torch.device
 model = CaptionModel(tokenizer)
 model.load_state_dict(weights['state_dict'])
 
+val_tfms = transforms.Compose([
+            # smaller edge of image resized to 256
+            transforms.Resize(256),
+            transforms.ToTensor(),
+            # normalize image for pre-trained model
+            transforms.Normalize((0.485, 0.456, 0.406),
+                                 (0.229, 0.224, 0.225))
+        ])
+
 def decode_caption(idxs, tokenizer):
     temp = []
     for i in idxs: