Hugging Face's logo

Spaces:
GbrlOl
/
MNIST
Sleeping

App Files Community
GbrlOl commited on
Commit
7a1ee5c
·
verified ·
1 Parent(s): c8dd1db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ model = CNN().to(device)
52
 
53
  # Cargar el modelo en la CPU
54
  model = CNN().to(device)
55
- model.load_state_dict(torch.load("model_mnist_cnn.pth", map_location=torch.device('cpu')))
56
 
57
  def predict(im):
58
 
 
52
 
53
  # Cargar el modelo en la CPU
54
  model = CNN().to(device)
55
+ model.load_state_dict(torch.load("model_mnist_cnn_data_augmentation.pth", map_location=torch.device('cpu')))
56
 
57
  def predict(im):
58