tudorgeorgescu commited on
Commit
63c4c37
·
verified ·
1 Parent(s): d829858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  # Load the keyphrase extraction models
5
- model1 = pipeline("ner", model="ml6team/keyphrase-extraction-kbir-inspec")
6
- model2 = pipeline("ner", model="aglazkova/bart_finetuned_keyphrase_extraction")
7
 
8
  # Streamlit app
9
  st.title("Keyphrase Extraction App")
 
2
  from transformers import pipeline
3
 
4
  # Load the keyphrase extraction models
5
+ model1 = pipeline("keyphrase-extraction", model="ml6team/keyphrase-extraction-kbir-inspec")
6
+ model2 = pipeline("keyphrase-extraction", model="aglazkova/bart_finetuned_keyphrase_extraction")
7
 
8
  # Streamlit app
9
  st.title("Keyphrase Extraction App")