Update app.py

app.py

@@ -9,7 +9,7 @@ from preprocess import Tokenizer, return_user_agent
 tokenizer = Tokenizer('./')
 tokenizer.load_tokenizer('./checkpoints/vocab-v1.pkl')
 
-weights = torch.load('./checkpoints/epoch=87-step=7144.ckpt', map_location=torch.device('cpu'))
+weights = torch.load('./checkpoints/caption_model.pt', map_location=torch.device('cpu'))
 model = CaptionModel(tokenizer)
 model.load_state_dict(weights['state_dict'])