Spaces:
Sleeping
Sleeping
updated app.py
Browse files
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,
|
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 =
|
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}")
|
|