Update app.py
Browse files
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="
|
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 |
|