Add local caption pipeline to UI plugin

2023-09-12 00:15:21 +03:00
parent 65b4551864
commit dc4a6e862b
3 changed files with 72 additions and 21 deletions
@@ -19,6 +19,8 @@ const securityOverride = false;
 const extras = {
    // Text classification model for sentiment analysis. HuggingFace ID of a model in ONNX format.
    classificationModel: 'Cohee/distilbert-base-uncased-go-emotions-onnx',
+    // Image captioning model. HuggingFace ID of a model in ONNX format.
+    captioningModel: 'Xenova/vit-gpt2-image-captioning',
 };

 // Request overrides for additional headers