Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
@@ -12,8 +12,8 @@ os.environ["HF_TOKEN"] = "your_huggingface_access_token"
|
|
12 |
|
13 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", use_auth_token=True)
|
14 |
whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny", use_auth_token=True)
|
15 |
-
text_model = AutoModelForSequenceClassification.from_pretrained("
|
16 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
17 |
|
18 |
def transcribe(audio_path):
|
19 |
waveform, sample_rate = torchaudio.load(audio_path)
|
|
|
12 |
|
13 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", use_auth_token=True)
|
14 |
whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny", use_auth_token=True)
|
15 |
+
text_model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
|
17 |
|
18 |
def transcribe(audio_path):
|
19 |
waveform, sample_rate = torchaudio.load(audio_path)
|