add arabic support

#1
by marsonal - opened
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,7 +36,7 @@ def transcribe(inputs, task, language, return_timestamps):
36
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
37
 
38
  # Map the language names to their corresponding codes
39
- language_codes = {"English": "en", "Korean": "ko", "Japanese": "ja"}
40
  language_code = language_codes.get(language, "en") # Default to "en" if the language is not found
41
  result = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task, "language": f"<|{language_code}|>"}, return_timestamps=return_timestamps)
42
 
@@ -118,7 +118,7 @@ mf_transcribe = gr.Interface(
118
  gr.inputs.Audio(source="microphone", type="filepath", optional=True),
119
  gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
120
  gr.inputs.Checkbox(label="Return timestamps"),
121
- gr.inputs.Dropdown(choices=["English", "Korean", "Japanese"], label="Language"),
122
  ],
123
  outputs="text",
124
  layout="horizontal",
@@ -137,7 +137,7 @@ file_transcribe = gr.Interface(
137
  gr.inputs.Audio(source="upload", type="filepath", optional=True, label="Audio file"),
138
  gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
139
  gr.inputs.Checkbox(label="Return timestamps"),
140
- gr.inputs.Dropdown(choices=["English", "Korean", "Japanese"], label="Language"),
141
  ],
142
  outputs="text",
143
  layout="horizontal",
 
36
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
37
 
38
  # Map the language names to their corresponding codes
39
+ language_codes = {"English": "en", "Korean": "ko", "Japanese": "ja", "Arabic":"ar"}
40
  language_code = language_codes.get(language, "en") # Default to "en" if the language is not found
41
  result = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task, "language": f"<|{language_code}|>"}, return_timestamps=return_timestamps)
42
 
 
118
  gr.inputs.Audio(source="microphone", type="filepath", optional=True),
119
  gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
120
  gr.inputs.Checkbox(label="Return timestamps"),
121
+ gr.inputs.Dropdown(choices=["English", "Korean", "Japanese", "Arabic"], label="Language"),
122
  ],
123
  outputs="text",
124
  layout="horizontal",
 
137
  gr.inputs.Audio(source="upload", type="filepath", optional=True, label="Audio file"),
138
  gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
139
  gr.inputs.Checkbox(label="Return timestamps"),
140
+ gr.inputs.Dropdown(choices=["English", "Korean", "Japanese", "Arabic"], label="Language"),
141
  ],
142
  outputs="text",
143
  layout="horizontal",