spuun commited on
Commit
33c7cc9
·
verified ·
1 Parent(s): 4a897ef

fix: update name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -133,11 +133,11 @@ class_names = [
133
  "Vegetation",
134
  ]
135
 
136
- best_model = ResNet101UNet(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__)), "3q7y4e.safetensors"),
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)