Update app.py

app.py

@@ -5,7 +5,22 @@ import fasttext
 
 model = fasttext.load_model(hf_hub_download("NbAiLab/nb-nordic-lid", "model.bin"))
 model_labels = set(label[-3:] for label in model.get_labels())
-
+language_dict = {
+    'dan': 'Danish',
+    'eng': 'English',
+    'fao': 'Faroese',
+    'fin': 'Finnish',
+    'isl': 'Icelandic',
+    'nno': 'Norwegian Nynorsk',
+    'nob': 'Norwegian Bokmål',
+    'sma': 'Southern Sami',
+    'sme': 'Northern Sami',
+    'smj': 'Lule Sami',
+    'smn': 'Inari Sami',
+    'sms': 'Skolt Sami',
+    'swe': 'Swedish',
+    'und': 'Undetermined',
+}
 
 def detect_lang(
     text: str,
@@ -54,7 +69,12 @@ def detect_lang(
 
 
 def identify(text):
-    return dict(detect_lang(text.replace("\n", " "), return_proba=True))
+    return {language_dict[lang]: proba for lang, proba in detect_lang(text.replace("\n", " "), return_proba=True)}
 
-iface = gr.Interface(fn=identify, inputs="text", outputs="label")
+iface = gr.Interface(
+    title="NB Nordic Language Identification",
+    description="""This demo uses the [NB-Nordic-LID](https://huggingface.co/NbAiLab/nb-nordic-lid) model to classify a given text into one of the 12 Nordic languages supported.""",
+    fn=identify,
+    inputs="text", 
+    outputs="label")
 iface.launch()