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 |
|