Spaces:
Runtime error
Runtime error
Commit
•
b04a665
1
Parent(s):
529b342
Update app.py
Browse files
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 =
|
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 |
|