Fix map_location issue

app.py

@@ -11,7 +11,7 @@ from huggingface_hub import hf_hub_download
 
 def check_simple_file(st_weights_path, torch_weights_path):
     st_weights = safetensors.torch.load_file(st_weights_path)
-    torch_weights = torch.load(torch_weights_path)
+    torch_weights = torch.load(torch_weights_path, map_location=torch.device('cpu'))
 
     # check if keys are the same
     if st_weights.keys() != torch_weights.keys():