IndranilB commited on
Commit
4d96d8c
1 Parent(s): eb3206d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -18
app.py CHANGED
@@ -99,29 +99,13 @@ def summarize():
99
  chunks = int(len_trans/512)
100
  from transformers import pipeline
101
 
102
- summarizer = pipeline("summarization", model="philschmid/flan-t5-base-samsum")
103
 
104
 
105
  #print(summarizer(transcript, do_sample=False))
106
  cutoff = 512
107
  final_output = ''
108
- """
109
- for i in range(chunks):
110
- print(i)
111
- if i == 0:
112
- tran_text = transcript[:512]
113
- inter_output = summarizer(tran_text, do_sample=False)[0]['summary_text']
114
- final_output += inter_output
115
- final_output += ' '
116
-
117
- else:
118
- #end_slice = cutoff + cutoff
119
- tran_text = transcript[cutoff:cutoff + 2]
120
- inter_output = summarizer(tran_text, do_sample=False)[0]['summary_text']
121
- final_output += inter_output
122
- final_output += ' '
123
- cutoff += cutoff
124
- """
125
  final_output = summarizer(transcript, do_sample=False)
126
  return final_output
127
 
 
99
  chunks = int(len_trans/512)
100
  from transformers import pipeline
101
 
102
+ summarizer = pipeline("summarization", model="Azma-AI/bart-large-text-summarizer")
103
 
104
 
105
  #print(summarizer(transcript, do_sample=False))
106
  cutoff = 512
107
  final_output = ''
108
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
109
  final_output = summarizer(transcript, do_sample=False)
110
  return final_output
111