Spaces:

paascorb
/

practica3

Runtime error

paascorb commited on Apr 9, 2023

Commit

5bbace9

1 Parent(s): 65f41de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import PIL
 import torchvision.transforms as transforms
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = torch.jit.load("unet.pth")
 model = model.cpu()
 def transform_image(image):

 import torchvision.transforms as transforms
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = torch.jit.load("prac3.pth")
 model = model.cpu()
 def transform_image(image):