Add default model for keyword generation

nlpodyssey · Oct 3, 2022 · 8209000 · 8209000
1 parent c6e6ac6
commit 8209000
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/tasks/text2text/text2text.go b/pkg/tasks/text2text/text2text.go
@@ -39,6 +39,10 @@ const (
 	// DefaultModelForAbstractiveQuestionAnswering is a summarization model fine-tuned for answer generation.
 	// Model card: https://huggingface.co/vblagoje/bart_lfqa/tree/main
 	DefaultModelForAbstractiveQuestionAnswering = "vblagoje/bart_lfqa"
+
+	// DefaultModelForKeywordsGeneration is a text generation model that produces a concatenated sequence of keyphrases.
+	// Model card: https://huggingface.co/bloomberg/KeyBART
+	DefaultModelForKeywordsGeneration = "bloomberg/KeyBART"
 )
 
 // DefaultModelForMachineTranslation specializes the model template for the source and target languages (iso-a2).