cfc-tech commited on
Commit
45c48be
·
verified ·
1 Parent(s): ce3a3a9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ if st.button('Summarize'):
36
  st.progress(50)
37
 
38
  # Speech to Text
39
- tokenizer = Wav2Vec2CTCTokenizer.from_pretrained("facebook/wav2vec2-base-960h")
40
- model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
41
 
42
  # Load and process the audio
43
  speech, _ = librosa.load(audio_path, sr=16000)
 
36
  st.progress(50)
37
 
38
  # Speech to Text
39
+ tokenizer = Wav2Vec2CTCTokenizer.from_pretrained("facebook/bart-large-cnn")
40
+ model = Wav2Vec2ForCTC.from_pretrained("facebook/bart-large-cnn")
41
 
42
  # Load and process the audio
43
  speech, _ = librosa.load(audio_path, sr=16000)