Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| transcription = pipeline("automatic-speech-recognition", model= "openai/whisper-base") | |
| clasification = pipeline( | |
| "audio-classification", | |
| model="anton-l/xtreme_s_xlsr_300m_minds14", | |
| ) | |
| def audio_a_text(audio): | |
| text = transcription(audio)["text"] | |
| return text | |
| def text_to_sentimient(audio): | |
| #text = transcription(audio)["text"] | |
| return clasification(audio) | |
| lang_id = { | |
| "Afrikaans": "af", | |
| "Amharic": "am", | |
| "Arabic": "ar", | |
| "Asturian": "ast", | |
| "Azerbaijani": "az", | |
| "Bashkir": "ba", | |
| "Belarusian": "be", | |
| "Bulgarian": "bg", | |
| "Bengali": "bn", | |
| "Breton": "br", | |
| "Bosnian": "bs", | |
| "Catalan": "ca", | |
| "Cebuano": "ceb", | |
| "Czech": "cs", | |
| "Welsh": "cy", | |
| "Danish": "da", | |
| "German": "de", | |
| "Greeek": "el", | |
| "English": "en", | |
| "Spanish": "es", | |
| "Estonian": "et", | |
| "Persian": "fa", | |
| "Fulah": "ff", | |
| "Finnish": "fi", | |
| "French": "fr", | |
| "Western Frisian": "fy", | |
| "Irish": "ga", | |
| "Gaelic": "gd", | |
| "Galician": "gl", | |
| "Gujarati": "gu", | |
| "Hausa": "ha", | |
| "Hebrew": "he", | |
| "Hindi": "hi", | |
| "Croatian": "hr", | |
| "Haitian": "ht", | |
| "Hungarian": "hu", | |
| "Armenian": "hy", | |
| "Indonesian": "id", | |
| "Igbo": "ig", | |
| "Iloko": "ilo", | |
| "Icelandic": "is", | |
| "Italian": "it", | |
| "Japanese": "ja", | |
| "Javanese": "jv", | |
| "Georgian": "ka", | |
| "Kazakh": "kk", | |
| "Central Khmer": "km", | |
| "Kannada": "kn", | |
| "Korean": "ko", | |
| "Luxembourgish": "lb", | |
| "Ganda": "lg", | |
| "Lingala": "ln", | |
| "Lao": "lo", | |
| "Lithuanian": "lt", | |
| "Latvian": "lv", | |
| "Malagasy": "mg", | |
| "Macedonian": "mk", | |
| "Malayalam": "ml", | |
| "Mongolian": "mn", | |
| "Marathi": "mr", | |
| "Malay": "ms", | |
| "Burmese": "my", | |
| "Nepali": "ne", | |
| "Dutch": "nl", | |
| "Norwegian": "no", | |
| "Northern Sotho": "ns", | |
| "Occitan": "oc", | |
| "Oriya": "or", | |
| "Panjabi": "pa", | |
| "Polish": "pl", | |
| "Pushto": "ps", | |
| "Portuguese": "pt", | |
| "Romanian": "ro", | |
| "Russian": "ru", | |
| "Sindhi": "sd", | |
| "Sinhala": "si", | |
| "Slovak": "sk", | |
| "Slovenian": "sl", | |
| "Somali": "so", | |
| "Albanian": "sq", | |
| "Serbian": "sr", | |
| "Swati": "ss", | |
| "Sundanese": "su", | |
| "Swedish": "sv", | |
| "Swahili": "sw", | |
| "Tamil": "ta", | |
| "Thai": "th", | |
| "Tagalog": "tl", | |
| "Tswana": "tn", | |
| "Turkish": "tr", | |
| "Ukrainian": "uk", | |
| "Urdu": "ur", | |
| "Uzbek": "uz", | |
| "Vietnamese": "vi", | |
| "Wolof": "wo", | |
| "Xhosa": "xh", | |
| "Yiddish": "yi", | |
| "Yoruba": "yo", | |
| "Chinese": "zh", | |
| "Zulu": "zu", | |
| } | |
| demo = gr.Blocks() | |
| with demo: | |
| gr.Markdown("Speech analyzer") | |
| audio = gr.Audio(type="filepath", label = "Upload a file") | |
| text = gr.Textbox() | |
| source_lang = gr.Dropdown(label="Source lang", choices=list(lang_id.keys())) | |
| #gr.Examples(examples = list(lang_id.keys()), | |
| # inputs=[ | |
| # source_lang]) | |
| b1 = gr.Button("convert to text") | |
| #b3 = gr.Button("select audio") | |
| #b3.Dropdown(inputs = source_lang) | |
| b1.click(audio_a_text, inputs=audio, outputs=text) | |
| b2 = gr.Button("Classification of speech") | |
| b2.click(text_to_sentimient, inputs=audio, outputs=text) | |
| demo.launch() | |