Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -87,7 +87,7 @@ def main():
|
|
87 |
elif choice=="Keyword Extractor":
|
88 |
st.subheader("Keyword Extraction")
|
89 |
#loading the pipeline
|
90 |
-
model_name = "yanekyuk/bert-uncased-keyword"
|
91 |
keyword_extractor = pipeline("text2text-generation", model=model_name, tokenizer=model_name)
|
92 |
input_text = st.text_area("Enter some text:")
|
93 |
|
|
|
87 |
elif choice=="Keyword Extractor":
|
88 |
st.subheader("Keyword Extraction")
|
89 |
#loading the pipeline
|
90 |
+
model_name = "yanekyuk/bert-uncased-keyword-extractor"
|
91 |
keyword_extractor = pipeline("text2text-generation", model=model_name, tokenizer=model_name)
|
92 |
input_text = st.text_area("Enter some text:")
|
93 |
|