Update app.py

app.py

@@ -271,7 +271,7 @@ class Music2emo:
         model = BTC_model(config=config.model).to(self.device)
 
         if os.path.isfile(model_file):
-            checkpoint = torch.load(model_file)
+            checkpoint = torch.load(model_file, map_location=self.device)
             mean = checkpoint['mean']
             std = checkpoint['std']
             model.load_state_dict(checkpoint['model'])