Update app.py

app.py

@@ -129,9 +129,11 @@ def get_adv_dif_model(latent_dim, output_dim):
 adv_dif_model = get_adv_dif_model(3, 128)
 adv_dif_model, _, _, _ = load_model_adv_dif(
     "adv_diff_2d_best.pt", 
-    adv_dif_model
+    adv_dif_model,
+    map_location=torch.device("cpu")
 )
 
+
 def generate_3d_visualization(Re, t_0, tau):
     dt = 2 / 500
     t = t_0 + tau * dt