com3dian commited on
Commit
98290ab
·
verified ·
1 Parent(s): e514b11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ if uploaded_file is not None:
65
  del Barttokenizer
66
  length = document.merge(25, 30, BartSE, device)
67
 
68
- summarizor = pipeline("summarization", model=summ_model_path, device = 0)
69
  summ_text = summarizor(document.segmentation['text'], max_length=100, min_length=10, do_sample=False)
70
  summ_text = [text['summary_text'] for text in summ_text]
71
 
 
65
  del Barttokenizer
66
  length = document.merge(25, 30, BartSE, device)
67
 
68
+ summarizor = pipeline("summarization", model=summ_model_path, device = device)
69
  summ_text = summarizor(document.segmentation['text'], max_length=100, min_length=10, do_sample=False)
70
  summ_text = [text['summary_text'] for text in summ_text]
71