Spaces:
Sleeping
Sleeping
spuuntries
commited on
Commit
·
0a3a9fc
1
Parent(s):
9cb0704
fix: abs path on model
Browse files
app.py
CHANGED
@@ -104,7 +104,10 @@ class_names = [
|
|
104 |
]
|
105 |
best_model = ResNet50(num_classes=len(class_names))
|
106 |
best_model = best_model.to(device)
|
107 |
-
load_model(
|
|
|
|
|
|
|
108 |
|
109 |
classifier = WasteClassifier(best_model, class_names, device)
|
110 |
|
|
|
104 |
]
|
105 |
best_model = ResNet50(num_classes=len(class_names))
|
106 |
best_model = best_model.to(device)
|
107 |
+
load_model(
|
108 |
+
best_model,
|
109 |
+
os.path.join(os.path.dirname(os.path.abspath(__file__)), "bjf8fp.safetensors"),
|
110 |
+
)
|
111 |
|
112 |
classifier = WasteClassifier(best_model, class_names, device)
|
113 |
|