benjolo commited on
Commit
acacecc
1 Parent(s): fa14146

Create run.py

Browse files
Files changed (1) hide show
  1. backend/run.py +36 -0
backend/run.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from transformers import pipeline
3
+
4
+ translator = pipeline("automatic-speech-recognition",
5
+ "facebook/seamless-m4t-v2-large",
6
+ torch_dtype=torch.float32,
7
+ device="cpu")
8
+
9
+ converter = pipeline("translation",
10
+ "facebook/seamless-m4t-v2-large",
11
+ torch_dtype=torch.float32,
12
+ device="cpu")
13
+
14
+
15
+ asr_text = translator("https://huggingface.co/datasets/reach-vb/random-audios/resolve/main/ted_60.wav",
16
+ chunk_length_s=30,
17
+ generate_kwargs={"tgt_lang": "eng"})
18
+
19
+ print(asr_text)
20
+
21
+ # print(translator("https://huggingface.co/datasets/reach-vb/random-audios/resolve/main/ted_60.wav",
22
+ # chunk_length_s=30,
23
+ # generate_kwargs={"tgt_lang": "fra"}))
24
+
25
+ # print(translator("https://huggingface.co/datasets/reach-vb/random-audios/resolve/main/ted_60.wav",
26
+ # chunk_length_s=30,
27
+ # generate_kwargs={"tgt_lang": "ita"}))
28
+
29
+ # print(translator("https://huggingface.co/datasets/reach-vb/random-audios/resolve/main/ted_60.wav",
30
+ # chunk_length_s=30,
31
+ # generate_kwargs={"tgt_lang": "spa"}))
32
+
33
+
34
+ print(converter(asr_text['text'],
35
+ src_lang="eng", tgt_lang="spa")
36
+ )