Fix extraction pipeline

app.py

@@ -12,7 +12,7 @@ from pipelines.keyphrase_generation_pipeline import KeyphraseGenerationPipeline
 @st.cache(allow_output_mutation=True, show_spinner=False)
 def load_pipeline(chosen_model):
     if "keyphrase-extraction" in chosen_model:
-        return KeyphraseExtractionPipeline(chosen_model, truncation=True)
+        return KeyphraseExtractionPipeline(chosen_model)
     elif "keyphrase-generation" in chosen_model:
         return KeyphraseGenerationPipeline(chosen_model, truncation=True)
 

pipelines/keyphrase_generation_pipeline.py

@@ -11,7 +11,7 @@ class KeyphraseGenerationPipeline(Text2TextGenerationPipeline):
     def __init__(self, model, keyphrase_sep_token=";", *args, **kwargs):
         super().__init__(
             model=AutoModelForSeq2SeqLM.from_pretrained(model),
-            tokenizer=AutoTokenizer.from_pretrained(model),
+            tokenizer=AutoTokenizer.from_pretrained(model, truncation=True),
             *args,
             **kwargs
         )