cfc-tech commited on
Commit
7912bc7
·
verified ·
1 Parent(s): 816a924

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from pytube import YouTube
3
  from moviepy.editor import *
4
  import torch
5
- from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
6
  from transformers import pipeline
7
  import librosa
8
 
@@ -33,8 +33,8 @@ if st.button('Summarize'):
33
  # Show progress
34
  st.progress(50)
35
 
36
- # Speech to Text
37
- tokenizer = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-base-960h")
38
  model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
39
 
40
  # Load and process the audio
@@ -63,4 +63,3 @@ if st.button('Summarize'):
63
 
64
  except Exception as e:
65
  st.error(f"An error occurred: {e}")
66
-
 
2
  from pytube import YouTube
3
  from moviepy.editor import *
4
  import torch
5
+ from transformers import Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer
6
  from transformers import pipeline
7
  import librosa
8
 
 
33
  # Show progress
34
  st.progress(50)
35
 
36
+ # Speech to Text using the corrected tokenizer
37
+ tokenizer = Wav2Vec2CTCTokenizer.from_pretrained("facebook/wav2vec2-base-960h")
38
  model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
39
 
40
  # Load and process the audio
 
63
 
64
  except Exception as e:
65
  st.error(f"An error occurred: {e}")