Spaces:
Runtime error
Runtime error
Commit
·
687b373
1
Parent(s):
37a3e4a
Update app.py
Browse files
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("
|
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
|