cotxetj commited on
Commit
3d57e67
1 Parent(s): a2519fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-fra")
21
  # Define a function to translate an audio, in english here
22
  def translate(audio):
23
  outputs = pipe(audio, max_new_tokens=256,
24
- generate_kwargs={"task": "transcribe", "language": "eng"})
25
  return outputs["text"]
26
 
27
 
@@ -76,7 +76,7 @@ demo = gr.Interface(
76
  gr.Audio(label="Record an audio", sources="microphone", type="numpy"),
77
  ],
78
  outputs=[
79
- gr.Text(label="Translation"),
80
  ],
81
  title=title,
82
  description=description,
 
21
  # Define a function to translate an audio, in english here
22
  def translate(audio):
23
  outputs = pipe(audio, max_new_tokens=256,
24
+ generate_kwargs={"task": "transcribe", "language": "english"})
25
  return outputs["text"]
26
 
27
 
 
76
  gr.Audio(label="Record an audio", sources="microphone", type="numpy"),
77
  ],
78
  outputs=[
79
+ gr.Text(label="Text translation")|gr.Audio(label="Audio translation"),
80
  ],
81
  title=title,
82
  description=description,