Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import json
-
+import os
+import string 
+import re
 import torch
 from transformers import pipeline
 from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
@@ -12,7 +14,7 @@ model_path = hf_hub_download(repo_id="cis-lmu/glotlid", filename="model.bin")
 identification_model = fasttext.load_model(model_path)
 def lang_ident(text):
     label, array = identification_model.predict(text)
-    label = get_name(label[0])
+    label = get_name(label[0].split('__')[-1].replace('_Hans', '_Hani').replace('_Hant', '_Hani'))
     return {language : label[0], socore : array[0]}
 
 pretrained_model: str = "facebook/m2m100_1.2B"