Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ asr_processor = Wav2Vec2Processor.from_pretrained("Baghdad99/saad-speech-recogni
|
|
7 |
|
8 |
# Load the Hausa translation model
|
9 |
translation_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/saad-hausa-text-to-english-text")
|
10 |
-
translation_model = AutoModelForSeq2SeqLM.from_pretrained("Baghdad99/saad-hausa-text-to-english-text")
|
11 |
|
12 |
# Load the Text-to-Speech model
|
13 |
tts_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/english_voice_tts")
|
|
|
7 |
|
8 |
# Load the Hausa translation model
|
9 |
translation_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/saad-hausa-text-to-english-text")
|
10 |
+
translation_model = AutoModelForSeq2SeqLM.from_pretrained("Baghdad99/saad-hausa-text-to-english-text", from_tf=True)
|
11 |
|
12 |
# Load the Text-to-Speech model
|
13 |
tts_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/english_voice_tts")
|