Spaces:
Sleeping
Sleeping
fix plant-disease-model
Browse files
app.py
CHANGED
|
@@ -90,7 +90,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
| 90 |
|
| 91 |
# Model yükleme
|
| 92 |
model = ResNet9(in_channels=3, num_diseases=len(CLASS_NAMES))
|
| 93 |
-
model.load_state_dict(torch.load('plant-disease-model
|
| 94 |
map_location=device,
|
| 95 |
weights_only=False))
|
| 96 |
model.to(device)
|
|
|
|
| 90 |
|
| 91 |
# Model yükleme
|
| 92 |
model = ResNet9(in_channels=3, num_diseases=len(CLASS_NAMES))
|
| 93 |
+
model.load_state_dict(torch.load('plant-disease-model.pth',
|
| 94 |
map_location=device,
|
| 95 |
weights_only=False))
|
| 96 |
model.to(device)
|
plant-disease-model-complete.pth → plant-disease-model.pth
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dd899ff558cae354f2e5465d3dcddd17ac87f5689014b0bf415df071cae8d240
|
| 3 |
+
size 26397109
|