Update app.py
Browse files
app.py
CHANGED
@@ -99,7 +99,7 @@ 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))
|
@@ -122,7 +122,7 @@ def summarize():
|
|
122 |
final_output += ' '
|
123 |
cutoff += cutoff
|
124 |
"""
|
125 |
-
final_output = summarizer(
|
126 |
return final_output
|
127 |
|
128 |
yt_link = st.text_input("Enter the YouTube URL: ")
|
|
|
99 |
chunks = int(len_trans/512)
|
100 |
from transformers import pipeline
|
101 |
|
102 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
103 |
|
104 |
|
105 |
#print(summarizer(transcript, do_sample=False))
|
|
|
122 |
final_output += ' '
|
123 |
cutoff += cutoff
|
124 |
"""
|
125 |
+
final_output = summarizer(transcript, do_sample=False)
|
126 |
return final_output
|
127 |
|
128 |
yt_link = st.text_input("Enter the YouTube URL: ")
|