Commit
•
1bb5dbb
1
Parent(s):
37da7d9
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,8 +2,8 @@ import streamlit as st
|
|
| 2 |
from transformers import pipeline,AutoProcessor, AutoTokenizer, AutoModelForTextToWaveform
|
| 3 |
|
| 4 |
# Load the audio classification model
|
| 5 |
-
processor = AutoProcessor.from_pretrained("
|
| 6 |
-
model = AutoModelForAudioClassification.from_pretrained("
|
| 7 |
|
| 8 |
# Load the TTS tokenizer and model
|
| 9 |
tts_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
|
|
|
|
| 2 |
from transformers import pipeline,AutoProcessor, AutoTokenizer, AutoModelForTextToWaveform
|
| 3 |
|
| 4 |
# Load the audio classification model
|
| 5 |
+
processor = AutoProcessor.from_pretrained("distilhubert-finetuned-stutterdetection")
|
| 6 |
+
model = AutoModelForAudioClassification.from_pretrained("distilhubert-finetuned-stutterdetection")
|
| 7 |
|
| 8 |
# Load the TTS tokenizer and model
|
| 9 |
tts_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
|