lpw commited on
Commit
f7a3b8e
1 Parent(s): 9fd9429

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -10
app.py CHANGED
@@ -1,14 +1,31 @@
1
- import gradio as gr
2
  import os
3
-
 
 
4
  import streamlit as st
5
 
6
- demo = gr.Interface.load(
7
- "huggingface/facebook/xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022",
8
- title="Speech-to-Speech",
9
- inputs="mic",
10
- description="Let me translate what you're saying!",
11
- api_key=st.secrets["api_key"]
12
- )
 
 
 
 
 
 
 
 
 
13
 
14
- demo.launch()
 
 
 
 
 
 
 
 
 
1
  import os
2
+ os.system("pip install gradio==3.3")
3
+ import gradio as gr
4
+ import numpy as np
5
  import streamlit as st
6
 
7
+ title = "Fairseq S2S"
8
+
9
+ description = "Gradio Demo for fairseq S2S: speech-to-speech translation models. To use it, simply add your audio, or click one of the examples to load them. Read more at the links below."
10
+
11
+ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2107.05604' target='_blank'>Direct speech-to-speech translation with discrete units</a> | <a href='https://github.com/facebookresearch/fairseq/tree/main/examples/speech_to_speech' target='_blank'>Github Repo</a></p>"
12
+
13
+ examples = [
14
+ ["enhanced_direct_s2st_units_audios_es-en_set2_source_12478_cv.flac","xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022"],
15
+ ]
16
+
17
+ io1 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022", api_key=st.secrets["api_key"])
18
+
19
+ def inference(text,model):
20
+ outtext = io1(text)
21
+ return outtext
22
+
23
 
24
+ gr.Interface(
25
+ inference,
26
+ [gr.inputs.Audio(label="Input",type="filepath"),gr.inputs.Dropdown(choices=["xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022"], type="value", label="model")
27
+ ],
28
+ gr.outputs.Audio(label="Output"),
29
+ article=article,
30
+ title=title,
31
+ description=description).queue().launch()