Spaces:

krafiq
/

Flexi-VAE

Sleeping

krafiq commited on Apr 16

Commit

c52fe79

verified ·

1 Parent(s): cc982c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -130,7 +130,7 @@ adv_dif_model = get_adv_dif_model(3, 128)
 adv_dif_model, _, _, _ = load_model_adv_dif(
     "adv_diff_2d_best.pt",
     adv_dif_model,
-    map_location=torch.device("cpu")
 )

 adv_dif_model, _, _, _ = load_model_adv_dif(
     "adv_diff_2d_best.pt",
     adv_dif_model,
+    map_location='cpu'
 )