Update app.py

app.py

@@ -21,6 +21,7 @@ from vinorm import TTSnorm
 
 nltk.download('vader_lexicon')
 os.system("python -m unidic download")
+os.system('nvidia-smi')
 
 HF_TOKEN = None
 api = HfApi(token=HF_TOKEN)
@@ -58,8 +59,8 @@ if torch.cuda.is_available():
 supported_languages = config.languages
 if not "vi" in supported_languages:
     supported_languages.append("vi")
-if not "es-ar" in supported_languages:
-    supported_languages.append("es-ar")
+if not "es-AR" in supported_languages:
+    supported_languages.append("es-AR")
 
 def normalize_vietnamese_text(text):
     text = (
@@ -234,7 +235,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
             language_gr = gr.Dropdown(
                 label="Idioma",
                 choices=[
-                    "es-ar",
+                    "es-AR",
                     "vi",
                     "en",
                     "es",
@@ -255,7 +256,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
                     "hi",
                 ],
                 max_choices=1,
-                value="es-ar",
+                value="es-AR",
             )
             normalize_text = gr.Checkbox(
                 label="Normalizar texto en vietnamita",