hellopahe commited on
Commit
d777f98
1 Parent(s): 4d1745d

add custom siblings

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py CHANGED
@@ -15,6 +15,14 @@ lex_l12 = LexRankL12()
15
 
16
  # 摘要方法1
17
  def extract_handler(content, siblings, num):
 
 
 
 
 
 
 
 
18
  summary_length = math.ceil(len(content) / 10)
19
  sentences = lex.find_central(content, siblings=siblings, num=num)
20
  output = ""
@@ -25,6 +33,14 @@ def extract_handler(content, siblings, num):
25
 
26
  # 摘要方法2
27
  def extract_handler_distiluse_v1(content, siblings, num):
 
 
 
 
 
 
 
 
28
  summary_length = math.ceil(len(content) / 10)
29
  sentences = lex_distiluse_v1.find_central(content, siblings=siblings, num=num)
30
  output = ""
@@ -35,6 +51,14 @@ def extract_handler_distiluse_v1(content, siblings, num):
35
 
36
  # 摘要方法3
37
  def extract_handler_l12(content, siblings, num):
 
 
 
 
 
 
 
 
38
  summary_length = math.ceil(len(content) / 10)
39
  sentences = lex_l12.find_central(content, siblings=siblings, num=num)
40
  output = ""
 
15
 
16
  # 摘要方法1
17
  def extract_handler(content, siblings, num):
18
+ if not siblings:
19
+ siblings = 0
20
+ if not num:
21
+ num = 10
22
+
23
+ siblings = int(siblings)
24
+ num = int(num)
25
+
26
  summary_length = math.ceil(len(content) / 10)
27
  sentences = lex.find_central(content, siblings=siblings, num=num)
28
  output = ""
 
33
 
34
  # 摘要方法2
35
  def extract_handler_distiluse_v1(content, siblings, num):
36
+ if not siblings:
37
+ siblings = 0
38
+ if not num:
39
+ num = 10
40
+
41
+ siblings = int(siblings)
42
+ num = int(num)
43
+
44
  summary_length = math.ceil(len(content) / 10)
45
  sentences = lex_distiluse_v1.find_central(content, siblings=siblings, num=num)
46
  output = ""
 
51
 
52
  # 摘要方法3
53
  def extract_handler_l12(content, siblings, num):
54
+ if not siblings:
55
+ siblings = 0
56
+ if not num:
57
+ num = 10
58
+
59
+ siblings = int(siblings)
60
+ num = int(num)
61
+
62
  summary_length = math.ceil(len(content) / 10)
63
  sentences = lex_l12.find_central(content, siblings=siblings, num=num)
64
  output = ""