Spaces:

wb-droid
/

Variational_Autoencoder

Runtime error

wb-droid commited on Jun 22, 2024

Commit

75403bb

1 Parent(s): 6974ee8

initial commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -130,7 +130,7 @@ labels_map = {
     9: "Ankle Boot",
 }
-myVAE2 = torch.load("checkpoint10.pt").to("cpu")
 myVAE2.eval()
 def sample():

     9: "Ankle Boot",
 }
+myVAE2 = torch.load("checkpoint10.pt", map_location=torch.device('cpu')).to("cpu")
 myVAE2.eval()
 def sample():