modified torch load

main.py

@@ -68,7 +68,7 @@ class EndpointHandler():
 		self.vocab = WordVocab(vocab_content)
 		self.trfm = TrfmSeq2seq(len(self.vocab), 256, len(self.vocab), 4)
 		device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-		self.trfm.load_state_dict(torch.load(trfm_path), map_location=device)
+		self.trfm.load_state_dict(torch.load(trfm_path, map_location=device))
 		self.trfm.eval()
 
 		# path to the pretrained models