Spaces:

ruescog
/

practica3

Runtime error

Rubén Escobedo commited on Mar 5, 2022

Commit

cc69e60

•

1 Parent(s): ea9face

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torchvision.transforms as transforms
 import torch
 # Cargamos el learner
-learn = load_model('best_model.pth')
 def transform_image(image, device):
     my_transforms = transforms.Compose([transforms.ToTensor(),

 import torch
 # Cargamos el learner
+learn = Learner.load('best_model')
 def transform_image(image, device):
     my_transforms = transforms.Compose([transforms.ToTensor(),