Update func_ai.py

func_ai.py

@@ -1,12 +1,13 @@
 import requests
 import torch
-from googletrans import Translator
+# from googletrans import Translator
 from transformers import pipeline
+from deep_translator import GoogleTranslator
 import time
 import os 
 VECTOR_API_URL = os.getenv('API_URL')
 
-translator = Translator()
+# translator = Translator()
 
 
 sentiment_model = pipeline(
@@ -26,7 +27,7 @@ classifier = pipeline(
 
 
 def classify_comment(text):
-    translated_text = translator.translate(text, dest='en').text
+    translated_text = GoogleTranslator(source='auto', target=target_language).translate(text)
     result = classifier(translated_text, ["interrogative", "non-interrogative"], clean_up_tokenization_spaces=True)
     top_class = result['labels'][0]
     return top_class