Montazerh82 commited on
Commit
a967799
·
verified ·
1 Parent(s): 53eeeb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -1,7 +1,23 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
 
 
 
 
 
 
 
 
 
4
  pipe = pipeline("text-classification", model="Mike0307/multilingual-e5-language-detection")
5
 
6
- demo = gr.Interface(fn=pipe, inputs="text", outputs="text")
 
 
 
 
 
 
7
  demo.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ languages = [
5
+ "Arabic", "Basque", "Breton", "Catalan", "Chinese_China", "Chinese_Hongkong",
6
+ "Chinese_Taiwan", "Chuvash", "Czech", "Dhivehi", "Dutch", "English",
7
+ "Esperanto", "Estonian", "French", "Frisian", "Georgian", "German", "Greek",
8
+ "Hakha_Chin", "Indonesian", "Interlingua", "Italian", "Japanese", "Kabyle",
9
+ "Kinyarwanda", "Kyrgyz", "Latvian", "Maltese", "Mongolian", "Persian", "Polish",
10
+ "Portuguese", "Romanian", "Romansh_Sursilvan", "Russian", "Sakha", "Slovenian",
11
+ "Spanish", "Swedish", "Tamil", "Tatar", "Turkish", "Ukranian", "Welsh"
12
+ ]
13
+
14
  pipe = pipeline("text-classification", model="Mike0307/multilingual-e5-language-detection")
15
 
16
+ def func(inp):
17
+ result = ''
18
+ for lang in inp:
19
+ result += languages[int(lang['label'][6:])] + ' ' + lang['score'] + '\n'
20
+ return result
21
+
22
+ demo = gr.Interface(fn=func, inputs="text", outputs="text")
23
  demo.launch()