Update app.py

app.py

@@ -82,7 +82,7 @@ def initiate_model(config, device):
     model = build_transformer(tokenizer_src.get_vocab_size(), tokenizer_tgt.get_vocab_size(), config["seq_len"], config['seq_len'], d_model=config['d_model']).to(device)
 
     model_filename = latest_weights_file_path(config)
-    state = torch.load(model_filename)
+    state = torch.load(model_filename, map_location=torch.device('cpu'))
     model.load_state_dict(state['model_state_dict'])
     return model, tokenizer_src, tokenizer_tgt