fedahumada commited on
Commit
9137e07
1 Parent(s): 76c2280

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -111
app.py CHANGED
@@ -2,110 +2,8 @@ import whisper
2
  import gradio as gr
3
  import time
4
  model = whisper.load_model("base")
5
- from transformers import pipeline
6
- es_en_translator = pipeline("translation_es_to_en")
7
-
8
- LANGUAGES = {
9
- "en": "english",
10
- "zh": "chinese",
11
- "de": "german",
12
- "es": "spanish",
13
- "ru": "russian",
14
- "ko": "korean",
15
- "fr": "french",
16
- "ja": "japanese",
17
- "pt": "portuguese",
18
- "tr": "turkish",
19
- "pl": "polish",
20
- "ca": "catalan",
21
- "nl": "dutch",
22
- "ar": "arabic",
23
- "sv": "swedish",
24
- "it": "italian",
25
- "id": "indonesian",
26
- "hi": "hindi",
27
- "fi": "finnish",
28
- "vi": "vietnamese",
29
- "iw": "hebrew",
30
- "uk": "ukrainian",
31
- "el": "greek",
32
- "ms": "malay",
33
- "cs": "czech",
34
- "ro": "romanian",
35
- "da": "danish",
36
- "hu": "hungarian",
37
- "ta": "tamil",
38
- "no": "norwegian",
39
- "th": "thai",
40
- "ur": "urdu",
41
- "hr": "croatian",
42
- "bg": "bulgarian",
43
- "lt": "lithuanian",
44
- "la": "latin",
45
- "mi": "maori",
46
- "ml": "malayalam",
47
- "cy": "welsh",
48
- "sk": "slovak",
49
- "te": "telugu",
50
- "fa": "persian",
51
- "lv": "latvian",
52
- "bn": "bengali",
53
- "sr": "serbian",
54
- "az": "azerbaijani",
55
- "sl": "slovenian",
56
- "kn": "kannada",
57
- "et": "estonian",
58
- "mk": "macedonian",
59
- "br": "breton",
60
- "eu": "basque",
61
- "is": "icelandic",
62
- "hy": "armenian",
63
- "ne": "nepali",
64
- "mn": "mongolian",
65
- "bs": "bosnian",
66
- "kk": "kazakh",
67
- "sq": "albanian",
68
- "sw": "swahili",
69
- "gl": "galician",
70
- "mr": "marathi",
71
- "pa": "punjabi",
72
- "si": "sinhala",
73
- "km": "khmer",
74
- "sn": "shona",
75
- "yo": "yoruba",
76
- "so": "somali",
77
- "af": "afrikaans",
78
- "oc": "occitan",
79
- "ka": "georgian",
80
- "be": "belarusian",
81
- "tg": "tajik",
82
- "sd": "sindhi",
83
- "gu": "gujarati",
84
- "am": "amharic",
85
- "yi": "yiddish",
86
- "lo": "lao",
87
- "uz": "uzbek",
88
- "fo": "faroese",
89
- "ht": "haitian creole",
90
- "ps": "pashto",
91
- "tk": "turkmen",
92
- "nn": "nynorsk",
93
- "mt": "maltese",
94
- "sa": "sanskrit",
95
- "lb": "luxembourgish",
96
- "my": "myanmar",
97
- "bo": "tibetan",
98
- "tl": "tagalog",
99
- "mg": "malagasy",
100
- "as": "assamese",
101
- "tt": "tatar",
102
- "haw": "hawaiian",
103
- "ln": "lingala",
104
- "ha": "hausa",
105
- "ba": "bashkir",
106
- "jw": "javanese",
107
- "su": "sundanese",
108
- }
109
 
110
  def transcribe(audio):
111
 
@@ -119,18 +17,18 @@ def transcribe(audio):
119
 
120
  # detect the spoken language
121
  _, probs = model.detect_language(mel)
122
- #print(f"Detected language: {max(probs, key=probs.get)}")
123
- lang = LANGUAGES[language]
124
- lang=(f"Detected language: {lang}")
125
 
126
 
127
  # decode the audio
128
  options = whisper.DecodingOptions(fp16 = False,task= "translate")
129
  result = whisper.decode(model, mel, options)
130
- word= result.text
131
- trans = es_en_translator(word)
132
- Trans = trans[0]['translation_text']
133
- result=f"{lang}\n{word}\n\nEnglish translation: {Trans}"
134
  return result.text
135
 
136
 
 
2
  import gradio as gr
3
  import time
4
  model = whisper.load_model("base")
5
+ #from transformers import pipeline
6
+ #es_en_translator = pipeline("translation_es_to_en")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
  def transcribe(audio):
9
 
 
17
 
18
  # detect the spoken language
19
  _, probs = model.detect_language(mel)
20
+ print(f"Detected language: {max(probs, key=probs.get)}")
21
+ #lang = LANGUAGES[language]
22
+ #lang=(f"Detected language: {lang}")
23
 
24
 
25
  # decode the audio
26
  options = whisper.DecodingOptions(fp16 = False,task= "translate")
27
  result = whisper.decode(model, mel, options)
28
+ #word= result.text
29
+ #trans = es_en_translator(word)
30
+ #Trans = trans[0]['translation_text']
31
+ #result=f"{lang}\n{word}\n\nEnglish translation: {Trans}"
32
  return result.text
33
 
34