arpm01 commited on
Commit
d1e5b5d
·
1 Parent(s): fb662eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -11
app.py CHANGED
@@ -1,15 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
3
-
4
- tokenizer = AutoTokenizer.from_pretrained("google/pegasus-xsum")
5
-
6
- model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-xsum")
7
-
8
- pipe = pipeline(task="summarization",
9
- model=model,
10
- tokenizer=tokenizer)
11
-
12
- gr.Interface.from_pipeline(gr.Interface.load("huggingface/google/pegasus-xsum"),
13
  title="Summarization using pegasus",
14
  description="Summarization description",
15
- allow_flagging="never").launch(inbrowser=True)
 
1
  import gradio as gr
2
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
3
+
4
+ gr.Interface.load("huggingface/google/pegasus-xsum",
 
 
 
 
 
 
 
 
5
  title="Summarization using pegasus",
6
  description="Summarization description",
7
+ ).launch()