Update app.py

app.py

@@ -96,7 +96,7 @@ if 'genstereo' not in globals():
 if 'fusion_model' not in globals():
     fusion_model = AdaptiveFusionLayer()
     fusion_checkpoint = join('checkpoints', CHECKPOINT_NAME, 'fusion_layer.pth')
-    fusion_model.load_state_dict(torch.load(fusion_checkpoint))
+    fusion_model.load_state_dict(torch.load(fusion_checkpoint, map_location='cpu'))
     fusion_model = fusion_model.to(DEVICE).eval()
 
 # Crop the image to the shorter side.