hamenend commited on
Commit
7647350
1 Parent(s): d6392c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,8 +1,15 @@
1
  import gradio as gr
 
 
 
 
 
2
 
3
  def greet(name):
4
  kw_model = KeyBERT()
5
- return kw_model.extract_keywords(name, keyphrase_ngram_range=(1, 1), stop_words=None,use_mmr=True,diversity=0.1)
 
 
6
 
7
  iface = gr.Interface(fn=greet, inputs="text", outputs="output")
8
  iface.launch(share=False)
 
1
  import gradio as gr
2
+ def getLabels(v):
3
+ arr = []
4
+ for tupla in keywords2:
5
+ arr.append(tupla[0])
6
+ return arr
7
 
8
  def greet(name):
9
  kw_model = KeyBERT()
10
+ analisis = kw_model.extract_keywords(name, keyphrase_ngram_range=(1, 1), stop_words=None,use_mmr=True,diversity=0.1)
11
+ claves = getLabels(analisis)
12
+ return ' '.join([str(elem) for elem in claves])
13
 
14
  iface = gr.Interface(fn=greet, inputs="text", outputs="output")
15
  iface.launch(share=False)