AWS Detect language

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import spacy
+from botocore.exceptions import ClientError
 from transformers import pipeline
 import boto3
 nlp = spacy.load("en_core_web_sm")
@@ -27,8 +28,16 @@ def greet(model_type, text):
             for entity in output]}
     elif model_type == "AWS Comprehend":
         client = boto3.client('comprehend')
+        try:
+            response = client.detect_dominant_language(Text=text)
+            languages = response['Languages']
+            print("Detected %s languages.", len(languages))
+            language = languages[0]['LanguageCode']
+        except ClientError:
+            print("Couldn't detect languages.")
+            language = 'en'
         response = client.detect_entities(
-                Text=text, LanguageCode='en')
+                Text=text, LanguageCode=language)
         print(response)
         return {"text": text, "entities": [{"word": entity["Text"], "entity": entity["Type"], "start": entity['BeginOffset'], 'end': entity['EndOffset']}
                                            for entity in response["Entities"]]}