fix deserialize weights in cpu only machine

app.py

@@ -28,7 +28,7 @@ cfg = load_params(cfg_path)
 def get_model(index):
     TransformerPath = model_paths[index]
     transformer = VQVAETransformer(cfg)
-    transformer.load_state_dict(torch.load(TransformerPath))
+    transformer.load_state_dict(torch.load(TransformerPath, map_location=device))
     transformer = transformer.to(device)
     transformer.eval()
     return transformer