IndranilB commited on
Commit
830562a
1 Parent(s): f9df875

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="snrspeaks/t5-one-line-summary")
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(tran_text, do_sample=False)
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: ")