add huggingface to catalog (#240)

alan-turing-institute · May 9, 2022 · 0f20950 · 0f20950
1 parent c33d93d
commit 0f20950
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/scivision/catalog/data/models.json b/scivision/catalog/data/models.json
@@ -103,6 +103,17 @@
             "labels_required":false,
             "institution":["alan-turing-institute"],
             "tags": ["classification", "2D", "image"]
+        },
+        {
+            "name":"scivision_huggingface",
+            "tasks":["classification"],
+            "url":"https://github.com/alan-turing-institute/scivision_huggingface",
+            "pkg_url":"git+https://github.com/alan-turing-institute/scivision_huggingface.git@main",
+            "format":"image",
+            "pretrained":true,
+            "labels_required":false,
+            "institution":["alan-turing-institute"],
+            "tags": ["classification", "2D", "image"]
         }
     ]
 }