virender74 commited on
Commit
687b373
·
1 Parent(s): 37a3e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -6,9 +6,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
6
 
7
  os.system("wget https://www.dropbox.com/s/grcragozd4x79zc/model_best.pth?dl=0")
8
 
9
- model = torch.load("/content/model_best.pth?dl=0", map_location=device)
10
-
11
- path = "/content/Potato-leaf-blight.jpeg"
12
 
13
  # img = Image.open(path).convert('RGB')
14
  from torchvision import transforms
 
6
 
7
  os.system("wget https://www.dropbox.com/s/grcragozd4x79zc/model_best.pth?dl=0")
8
 
9
+ model = torch.load("./model_best.pth?dl=0", map_location=device)
 
 
10
 
11
  # img = Image.open(path).convert('RGB')
12
  from torchvision import transforms