x

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import pipeline
 
 # Load the keyphrase extraction models
-model1 = pipeline("keyphrase-extraction", model="ml6team/keyphrase-extraction-kbir-inspec")
-model2 = pipeline("keyphrase-extraction", model="aglazkova/bart_finetuned_keyphrase_extraction")
+model1 = pipeline("text2text-generation", model="ml6team/keyphrase-extraction-kbir-inspec")
+model2 = pipeline("text2text-generation", model="aglazkova/bart_finetuned_keyphrase_extraction")
 
 # Streamlit app
 st.title("Keyphrase Extraction App")