detectron2-model-demo

Running

amaurisarchibald commited on Feb 9, 2024

Commit

c712b7e

verified ·

1 Parent(s): 5e79a3f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ models = [
 },
 {
     "name": "Version 2 (4-class)",
-    "model_path": "https://huggingface.co/amaurisarchibald/model_damage/resolve/main/model_final.pth",
     "classes": ["damage", "scratch"],
     "cfg": None,
     "metadata": None

 },
 {
     "name": "Version 2 (4-class)",
+    "model_path": "https://huggingface.co/amaurisarchibald/model_damage/resolve/main/model_damage.pth",
     "classes": ["damage", "scratch"],
     "cfg": None,
     "metadata": None