IotaCluster commited on
Commit
ed33f5c
·
verified ·
1 Parent(s): 4e9cc7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ summarizer = pipeline(
12
  "summarization",
13
  model=model,
14
  tokenizer=tokenizer,
15
- device=-1 # set to -1 for CPU, or the GPU ID if available
 
16
  )
17
 
18
  def remove_confidentiality(text: str) -> str:
 
12
  "summarization",
13
  model=model,
14
  tokenizer=tokenizer,
15
+ device=-1, # set to -1 for CPU, or the GPU ID if available
16
+ truncation = True
17
  )
18
 
19
  def remove_confidentiality(text: str) -> str: