spuuntries commited on
Commit
0a3a9fc
·
1 Parent(s): 9cb0704

fix: abs path on model

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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(best_model, os.path.join(__file__, "..", "bjf8fp.safetensors"))
 
 
 
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