Wootang01 commited on
Commit
718f074
1 Parent(s): eb656bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  from gradio.mix import Parallel, Series
 
3
 
4
  title = "Extractive Text Summarizer"
5
  description = "Past an article text or other text. Submit the text and the machine will create four summaries based on words in the text. Which sentences in the text are the most important for the summaries? Which summaries are better for your case?"
@@ -39,8 +40,8 @@ It also noted that JUPAS updating its program list may also contribute to the dr
39
 
40
  io1 = gr.Interface.load('huggingface/sshleifer/distilbart-cnn-12-6')
41
  io2 = gr.Interface.load("huggingface/facebook/bart-large-cnn")
42
- io3 = gr.Interface.load("huggingface/google/pegasus-xsum")
43
- io4 = gr.Interface.load("huggingface/sshleifer/distilbart-cnn-6-6")
44
 
45
  iface = Parallel(io1, io2, io3, io4,
46
  theme='huggingface',
 
1
  import gradio as gr
2
  from gradio.mix import Parallel, Series
3
+ from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForSeq2SeqLM
4
 
5
  title = "Extractive Text Summarizer"
6
  description = "Past an article text or other text. Submit the text and the machine will create four summaries based on words in the text. Which sentences in the text are the most important for the summaries? Which summaries are better for your case?"
 
40
 
41
  io1 = gr.Interface.load('huggingface/sshleifer/distilbart-cnn-12-6')
42
  io2 = gr.Interface.load("huggingface/facebook/bart-large-cnn")
43
+ io3 = gr.Interface.load("huggingface/t5-large")
44
+ io4 = gr.Interface.load("huggingface/t5-base")
45
 
46
  iface = Parallel(io1, io2, io3, io4,
47
  theme='huggingface',