Spaces:
Sleeping
Sleeping
fix: update name
Browse files
app.py
CHANGED
|
@@ -133,11 +133,11 @@ class_names = [
|
|
| 133 |
"Vegetation",
|
| 134 |
]
|
| 135 |
|
| 136 |
-
best_model =
|
| 137 |
best_model = best_model.to(device)
|
| 138 |
load_model(
|
| 139 |
best_model,
|
| 140 |
-
os.path.join(os.path.dirname(os.path.abspath(__file__)), "
|
| 141 |
)
|
| 142 |
|
| 143 |
classifier = WasteClassifier(best_model, class_names, device)
|
|
|
|
| 133 |
"Vegetation",
|
| 134 |
]
|
| 135 |
|
| 136 |
+
best_model = ResNet18UNet(num_classes=len(class_names))
|
| 137 |
best_model = best_model.to(device)
|
| 138 |
load_model(
|
| 139 |
best_model,
|
| 140 |
+
os.path.join(os.path.dirname(os.path.abspath(__file__)), "ppn62p.safetensors"),
|
| 141 |
)
|
| 142 |
|
| 143 |
classifier = WasteClassifier(best_model, class_names, device)
|