Spaces:

GV05
/

Variational-Autoencoder-on-MNIST

Runtime error

GV05 commited on Oct 4, 2022

Commit

ce00128

1 Parent(s): 093675e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ H_DIM = 512
 Z_DIM = 256
 model = VariationalAutoEncoder(INPUT_DIM, H_DIM, Z_DIM)
-model.load_state_dict(torch.load("MnistVAEmodel.pth"))
 model.eval()
 def predict(img):
     img = img.convert('1')

 Z_DIM = 256
 model = VariationalAutoEncoder(INPUT_DIM, H_DIM, Z_DIM)
+model.load_state_dict(torch.load("MnistVAEmodel.pt"))
 model.eval()
 def predict(img):
     img = img.convert('1')