diff --git a/docs/scripts/make_model_zoo_docs.py b/docs/scripts/make_model_zoo_docs.py index 472e0b31db..67769c5e90 100644 --- 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 }}")