Spaces:
Sleeping
Sleeping
File size: 1,463 Bytes
8032379 d5e322f af2b7f7 8032379 d5e322f af2b7f7 d5e322f e901c60 d5e322f e901c60 37ec07c e901c60 d5e322f f9f6585 946c41f d5e322f 946c41f d5e322f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import gradio as gr
from kwextractor import KeyWordExtractor
import csv, io
kw_ex=KeyWordExtractor()
csv_encoded=io.StringIO()
writer = csv.writer(csv_encoded)
def generate_kws(context,num_kw, kw_ngs):
context=context.strip()
if context:
try:
num_kw=int(num_kw)
except ValueError:
num_kw=None
try:
kw_ngs=int(kw_ngs)
except ValueError:
kw_ngs=None
#csv_encoded.truncate(0)
#writer.writerow([context])
#context=csv_encoded.getvalue()
return kw_ex.extract(context, num_kw, kw_ngs) or ""
else:
raise gr.Error("Please enter text in inputbox!!!!")
inputs=gr.Textbox(value="", lines=5, label="Input Context",elem_id="inp_div")
nkws = gr.Textbox(label="Number of keywords to extract",default="3",elem_id="inp_div")
kw_ngs= gr.Textbox(label="Maximum number of ngrams per keyword",default="3",elem_id="inp_div")
outputs=gr.Textbox(label="Generated Keywords",lines=6,elem_id="inp_div")
demo = gr.Interface(
generate_kws,
[inputs,nkws,kw_ngs],
outputs,
title="Keyword Extraction Model",
css=".gradio-container {background-color: lightgray} #inp_div {background-color: #7FB3D5;}",
article="""<p style='text-align: center;'>Feel free to give us your <a href="https://www.pragnakalp.com/contact/" target="_blank">feedback</a> on this Keyword Extraction demo.</p>
"""
)
demo.launch()
|