fix

app.py

@@ -1,9 +1,14 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline, AutoModelForSeq2SeqLM
 
 # Load the keyphrase extraction models
-model1 = pipeline("text2text-generation", model="ml6team/keyphrase-extraction-kbir-inspec")
-model2 = pipeline("text2text-generation", model="aglazkova/bart_finetuned_keyphrase_extraction")
+tokenizer1 = AutoTokenizer.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec")
+model1 = AutoModelForTokenClassification.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec")
+extractor1 = pipeline("ner", model=model1, tokenizer=tokenizer1)
+
+tokenizer2 = AutoTokenizer.from_pretrained("aglazkova/bart_finetuned_keyphrase_extraction")
+model2 = AutoModelForSeq2SeqLM.from_pretrained("aglazkova/bart_finetuned_keyphrase_extraction")
+extractor2 = pipeline("text2text-generation", model=model2, tokenizer=tokenizer2)
 
 # Streamlit app
 st.title("Keyphrase Extraction App")
@@ -15,15 +20,17 @@ user_input = st.text_area("Text Input", "")
 # Extract keyphrases
 if st.button("Extract Keyphrases"):
     if user_input:
-        keyphrases_model1 = model1(user_input)
-        keyphrases_model2 = model2(user_input)
-
+        # Extract keyphrases using the first model
+        keyphrases_model1 = extractor1(user_input)
         st.write("Extracted Keyphrases from `ml6team/keyphrase-extraction-kbir-inspec`:")
         for keyphrase in keyphrases_model1:
             st.write(keyphrase['word'])
 
+        # Extract keyphrases using the second model
+        tokenized_text = tokenizer2.prepare_seq2seq_batch([user_input], return_tensors='pt')
+        translation = model2.generate(**tokenized_text)
+        translated_text = tokenizer2.batch_decode(translation, skip_special_tokens=True)[0]
         st.write("Extracted Keyphrases from `aglazkova/bart_finetuned_keyphrase_extraction`:")
-        for keyphrase in keyphrases_model2:
-            st.write(keyphrase['word'])
+        st.write(translated_text)
     else:
         st.write("Please enter some text to extract keyphrases.")