amielle commited on
Commit
24df5eb
β€’
1 Parent(s): 6ee1cee

chore: Update models with paper and code sources

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -14,7 +14,6 @@ iface = gr.Interface(
14
  placeholder="US9820315B2 or https://patents.google.com/patent/US9820315B2 ...",
15
  lines=1,
16
  default="US9820315B2"),
17
- # gr.inputs.Radio(["Link", "Patent Code"]), # Can be figured out automatically via parsing
18
  gr.inputs.CheckboxGroup(summarizer.summary_options,
19
  default=summarizer.summary_options,
20
  label="Summaries to Generate"),
@@ -32,7 +31,7 @@ iface = gr.Interface(
32
  optional=False),
33
  gr.inputs.Slider(minimum=250,
34
  maximum=1000,
35
- step=5,
36
  default=250,
37
  label="Input Document Word Limit"),
38
  ],
@@ -59,10 +58,18 @@ iface = gr.Interface(
59
  πŸ“‘ Returns summaries for the abstract, background, and/or claims.
60
 
61
  Models explored:
62
- - https://huggingface.co/google/bigbird-pegasus-large-bigpatent
63
- - https://huggingface.co/cnicu/t5-small-booksum
64
- - https://huggingface.co/sshleifer/distilbart-cnn-6-6
65
- - https://huggingface.co/google/pegasus-xsum
 
 
 
 
 
 
 
 
66
  """
67
  )
68
 
 
14
  placeholder="US9820315B2 or https://patents.google.com/patent/US9820315B2 ...",
15
  lines=1,
16
  default="US9820315B2"),
 
17
  gr.inputs.CheckboxGroup(summarizer.summary_options,
18
  default=summarizer.summary_options,
19
  label="Summaries to Generate"),
 
31
  optional=False),
32
  gr.inputs.Slider(minimum=250,
33
  maximum=1000,
34
+ step=10,
35
  default=250,
36
  label="Input Document Word Limit"),
37
  ],
 
58
  πŸ“‘ Returns summaries for the abstract, background, and/or claims.
59
 
60
  Models explored:
61
+ - Big Bird: Transformers for Longer Sequences
62
+ https://arxiv.org/abs/2007.14062
63
+ https://huggingface.co/google/bigbird-pegasus-large-bigpatent
64
+ - T5
65
+ https://arxiv.org/pdf/1910.10683.pdf
66
+ https://huggingface.co/cnicu/t5-small-booksum
67
+ - BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehension
68
+ https://arxiv.org/abs/1910.13461
69
+ https://huggingface.co/sshleifer/distilbart-cnn-6-6
70
+ - PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarization
71
+ https://arxiv.org/abs/1912.08777
72
+ https://huggingface.co/google/pegasus-xsum
73
  """
74
  )
75