samuelleecong commited on
Commit
b04a665
1 Parent(s): 529b342

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
17
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device, generate_kwargs = {"task": "translate"})
18
 
19
  model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
20
- tokenizer = M2M100Tokenizer.from_pretrained("alirezamsh/small100", tgt_lang="sw")
21
 
22
 
23
 
 
17
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device, generate_kwargs = {"task": "translate"})
18
 
19
  model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
20
+ tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100", tgt_lang="sw")
21
 
22
 
23