Spaces:

Baghdad99
/

ha-en

Sleeping

Baghdad99 commited on Dec 6, 2023

Commit

d9741d3

1 Parent(s): 5dd0786

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ translation_model = AutoModelForSeq2SeqLM.from_pretrained("Baghdad99/saad-hausa-
 tts_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/english_voice_tts")
 tts_model = AutoModelForTextToWaveform.from_pretrained("Baghdad99/english_voice_tts")
-# Define the translation and synthesis functions
 def translate(audio_signal, sampling_rate):
     inputs = asr_processor(audio_signal, return_tensors="pt", padding=True, sampling_rate=sampling_rate)
     logits = asr_model(inputs.input_values).logits

 tts_tokenizer = AutoTokenizer.from_pretrained("Baghdad99/english_voice_tts")
 tts_model = AutoModelForTextToWaveform.from_pretrained("Baghdad99/english_voice_tts")
+# Modify the translate function to accept the sampling_rate argument
 def translate(audio_signal, sampling_rate):
     inputs = asr_processor(audio_signal, return_tensors="pt", padding=True, sampling_rate=sampling_rate)
     logits = asr_model(inputs.input_values).logits