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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-xsum")
8
  pipe = pipeline(task="summarization",
9
  model=model,
10
  tokenizer=tokenizer)
11
- gr.Interface.from_pipeline(pipe,
 
12
  title="Summarization using pegasus",
13
  description="Summarization description",
14
  allow_flagging="never").launch(inbrowser=True)
 
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)