madankn79 commited on
Commit
0c65a3b
·
1 Parent(s): 73fdc8f

Initial Version

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ from transformers_js_py import pipeline
3
 
4
  pipe = await pipeline("summarization", model="facebook/bart-large-cnn")
5
 
 
6
  async def process(text):
7
- return await pipe(text)
8
 
9
  demo = gr.Interface(fn=process, inputs="text", outputs="json")
10
 
 
3
 
4
  pipe = await pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
+
7
  async def process(text):
8
+ return await pipe(ARTICLE, max_length=65, min_length=30, do_sample=False)
9
 
10
  demo = gr.Interface(fn=process, inputs="text", outputs="json")
11