SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add video classification pipeline (#20151)

Browse Source 
* 🚧 wip video classification pipeline

* 🚧 wip - add is_decord_available check

* 🐛 add missing import

* ✅ add tests

* 🔧 add decord to setup extras

* 🚧 add is_decord_available

* ✨ add video-classification pipeline

* 📝 add video classification pipe to docs

* 🐛 add missing VideoClassificationPipeline import

* 📌 add decord install in test runner

* ✅ fix url inputs to video-classification pipeline

* ✨ updates from review

* 📝 add video cls pipeline to docs

* 📝 add docstring

* 🔥 remove unused import

* 🔥 remove some code

* 📝 docfix
This commit is contained in:
Nathan Raw
2022-12-08 16:22:43 -05:00
committed by GitHub
parent c56ebbbea6
commit 9e56aff58a
12 changed files with 272 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
utils/update_metadata.py
View File

@@ -102,6 +102,7 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [
"AutoModel",
),
("depth-estimation", "MODEL_FOR_DEPTH_ESTIMATION_MAPPING_NAMES", "AutoModelForDepthEstimation"),
("video-classification", "MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING_NAMES", "AutoModelForVideoClassification"),
]