Spaces:

vinni1484
/

text-keywords

Runtime error

text-keywords / app.py

Update app.py

39c0fbb almost 2 years ago

No virus

358 Bytes

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