Spaces:
Sleeping
Sleeping
medmediani
commited on
Commit
·
af2b7f7
1
Parent(s):
eb28344
Changed the model path
Browse files- .ipynb_checkpoints/app-checkpoint.py +6 -0
- .ipynb_checkpoints/kwextractor-checkpoint.py +1 -1
- app.py +6 -0
- kwextractor.py +1 -1
.ipynb_checkpoints/app-checkpoint.py
CHANGED
@@ -1,8 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from kwextractor import KeyWordExtractor
|
|
|
3 |
|
4 |
|
5 |
kw_ex=KeyWordExtractor()
|
|
|
|
|
6 |
|
7 |
|
8 |
def generate_kws(context,num_kw, kw_ngs):
|
@@ -17,6 +20,9 @@ def generate_kws(context,num_kw, kw_ngs):
|
|
17 |
kw_ngs=int(kw_ngs)
|
18 |
except ValueError:
|
19 |
kw_ngs=None
|
|
|
|
|
|
|
20 |
return kw_ex.extract(context, num_kw, kw_ngs) or ""
|
21 |
else:
|
22 |
raise gr.Error("Please enter text in inputbox!!!!")
|
|
|
1 |
import gradio as gr
|
2 |
from kwextractor import KeyWordExtractor
|
3 |
+
import csv, io
|
4 |
|
5 |
|
6 |
kw_ex=KeyWordExtractor()
|
7 |
+
csv_encoded=io.StringIO()
|
8 |
+
writer = csv.writer(csv_encoded)
|
9 |
|
10 |
|
11 |
def generate_kws(context,num_kw, kw_ngs):
|
|
|
20 |
kw_ngs=int(kw_ngs)
|
21 |
except ValueError:
|
22 |
kw_ngs=None
|
23 |
+
csv_encoded.truncate(0)
|
24 |
+
writer.writerow([context])
|
25 |
+
context=csv_encoded.getvalue()
|
26 |
return kw_ex.extract(context, num_kw, kw_ngs) or ""
|
27 |
else:
|
28 |
raise gr.Error("Please enter text in inputbox!!!!")
|
.ipynb_checkpoints/kwextractor-checkpoint.py
CHANGED
@@ -34,7 +34,7 @@ class KeyWordExtractor():
|
|
34 |
top_n=nkws,
|
35 |
#use_maxsum=True,nr_candidates=20, top_n=5,
|
36 |
#use_mmr=True,
|
37 |
-
diversity=0.8,
|
38 |
stop_words=None)
|
39 |
return ", ".join(w for w,_ in kw)
|
40 |
|
|
|
34 |
top_n=nkws,
|
35 |
#use_maxsum=True,nr_candidates=20, top_n=5,
|
36 |
#use_mmr=True,
|
37 |
+
#diversity=0.8,
|
38 |
stop_words=None)
|
39 |
return ", ".join(w for w,_ in kw)
|
40 |
|
app.py
CHANGED
@@ -1,8 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from kwextractor import KeyWordExtractor
|
|
|
3 |
|
4 |
|
5 |
kw_ex=KeyWordExtractor()
|
|
|
|
|
6 |
|
7 |
|
8 |
def generate_kws(context,num_kw, kw_ngs):
|
@@ -17,6 +20,9 @@ def generate_kws(context,num_kw, kw_ngs):
|
|
17 |
kw_ngs=int(kw_ngs)
|
18 |
except ValueError:
|
19 |
kw_ngs=None
|
|
|
|
|
|
|
20 |
return kw_ex.extract(context, num_kw, kw_ngs) or ""
|
21 |
else:
|
22 |
raise gr.Error("Please enter text in inputbox!!!!")
|
|
|
1 |
import gradio as gr
|
2 |
from kwextractor import KeyWordExtractor
|
3 |
+
import csv, io
|
4 |
|
5 |
|
6 |
kw_ex=KeyWordExtractor()
|
7 |
+
csv_encoded=io.StringIO()
|
8 |
+
writer = csv.writer(csv_encoded)
|
9 |
|
10 |
|
11 |
def generate_kws(context,num_kw, kw_ngs):
|
|
|
20 |
kw_ngs=int(kw_ngs)
|
21 |
except ValueError:
|
22 |
kw_ngs=None
|
23 |
+
csv_encoded.truncate(0)
|
24 |
+
writer.writerow([context])
|
25 |
+
context=csv_encoded.getvalue()
|
26 |
return kw_ex.extract(context, num_kw, kw_ngs) or ""
|
27 |
else:
|
28 |
raise gr.Error("Please enter text in inputbox!!!!")
|
kwextractor.py
CHANGED
@@ -34,7 +34,7 @@ class KeyWordExtractor():
|
|
34 |
top_n=nkws,
|
35 |
#use_maxsum=True,nr_candidates=20, top_n=5,
|
36 |
#use_mmr=True,
|
37 |
-
diversity=0.8,
|
38 |
stop_words=None)
|
39 |
return ", ".join(w for w,_ in kw)
|
40 |
|
|
|
34 |
top_n=nkws,
|
35 |
#use_maxsum=True,nr_candidates=20, top_n=5,
|
36 |
#use_mmr=True,
|
37 |
+
#diversity=0.8,
|
38 |
stop_words=None)
|
39 |
return ", ".join(w for w,_ in kw)
|
40 |
|