Fixed example for zero-shot-detection-transformer-torch

voxel51 · Dec 5, 2024 · 06cab9a · 06cab9a
1 parent 1649f2f
commit 06cab9a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/docs/scripts/make_model_zoo_docs.py b/docs/scripts/make_model_zoo_docs.py
@@ -173,6 +173,17 @@
     model = foz.load_zoo_model("{{ name }}")
 
     embeddings = dataset.compute_embeddings(model)
+
+{% elif 'zero-shot-detection-transformer-torch' in name %}
+    model = foz.load_zoo_model(
+    "{{ name }}",
+    classes=["person", "dog", "cat", "bird", "car", "tree", "chair"]
+    )
+
+    dataset.apply_model(model, label_field="predictions")
+
+    session = fo.launch_app(dataset)
+    
 {% else %}
     model = foz.load_zoo_model("{{ name }}")