Hugging Face's logo

Spaces:
preetam8
/
speech-to-speech-translation
Sleeping

App Files Community
preetam8 commited on
Commit
472a003
·
1 Parent(s): 0f4883a

fix device

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ target_language = "fr"
14
 
15
  # load speech translation checkpoint
16
  asr_pipe = pipeline("automatic-speech-recognition", model="bofenghuang/whisper-small-cv11-french", device=device)
17
- translation_model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100", device=device)
18
  translation_tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100", tgt_lang=target_language)
19
 
20
  # load text-to-speech checkpoint
 
14
 
15
  # load speech translation checkpoint
16
  asr_pipe = pipeline("automatic-speech-recognition", model="bofenghuang/whisper-small-cv11-french", device=device)
17
+ translation_model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
18
  translation_tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100", tgt_lang=target_language)
19
 
20
  # load text-to-speech checkpoint