AdamNovotnyCom commited on
Commit
5574f70
1 Parent(s): 55c8e0a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ elif "summary" == os.environ.get("MODEL"):
52
  model="facebook/bart-large-cnn"
53
  summarizer = transformers.pipeline("summarization", model=model)
54
  def model_func(input_text, request: gr.Request):
55
- return summarizer(input_text, max_length=300, min_length=30, do_sample=False)[0]["summary_text"]
56
 
57
  input_label = "How can I help?"
58
  if "summary" == os.environ.get("MODEL"):
 
52
  model="facebook/bart-large-cnn"
53
  summarizer = transformers.pipeline("summarization", model=model)
54
  def model_func(input_text, request: gr.Request):
55
+ return summarizer(input_text, max_length=130, min_length=30, do_sample=False)[0]["summary_text"]
56
 
57
  input_label = "How can I help?"
58
  if "summary" == os.environ.get("MODEL"):