Hugging Face's logo

Spaces:
debojit01
/
mnist
Sleeping

App Files Community
debojit01 commited on
Commit
4fb7754
·
verified ·
1 Parent(s): 78eb968

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ class VAE(torch.nn.Module):
34
 
35
  # Load model
36
  model = VAE()
37
- model.load_state_dict(torch.load("vae_mnist.pth", map_location='cpu'))
38
  model.eval()
39
 
40
  # Generation function for Gradio
 
34
 
35
  # Load model
36
  model = VAE()
37
+ model.load_state_dict(torch.load("cvae_mnist.pth", map_location='cpu'))
38
  model.eval()
39
 
40
  # Generation function for Gradio