paascorb commited on
Commit
706b9bb
1 Parent(s): 5bbace9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import PIL
9
  import torchvision.transforms as transforms
10
 
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
- model = torch.jit.load("prac3.pth")
13
  model = model.cpu()
14
 
15
  def transform_image(image):
 
9
  import torchvision.transforms as transforms
10
 
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
+ model = torch.jit.load("pract3.pth")
13
  model = model.cpu()
14
 
15
  def transform_image(image):