All the models available in the Generative AI Lab are listed in this page. The models are either trained within the Generative AI Lab, uploaded to Generative AI Lab by admin users, or downloaded from NLP Models Hub. General information about the models like labels/categories and the source (downloaded/trained/uploaded) is viewable. It is possible to delete any model, or redownload failed ones from the options available under the more action menu on each model.
All available models are listed in the Spark NLP Pipeline Config on the Setup Page of any project and are ready to be included in the Labeling Config for pre-annotation.
Auto download of model dependencies
Starting from version 2.8.0, Generative AI Lab automatically downloads all the necessary dependencies along with the model saving users valuable time from manually downloading the dependencies. Previously, users had to first download the model from the Models Hub page (e.g. ner_healthcare_de) and then again download the necessary embeddings required to train the model (e.g. w2v_cc_300d).
Custom Model Upload
Custom models can be uploaded using the Upload button present in the top right corner of the page. The labels predicted by this model need to be specified in the upload form.
Note: The models to upload need to be Spark NLP compatible.