Spaces:
Runtime error
Runtime error
import gradio as gr | |
from keybert import KeyBERT | |
model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1") | |
def keywords(text): | |
keywords = model.extract_keywords(text, keyphrase_ngram_range=(1,2), top_n=10) | |
keywords = dict(keywords) | |
return keywords | |
gr.Interface(keywords, "text", "text", title="Keyword Extractor").launch() |