Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -168,15 +168,19 @@ class TranslationBase(ABC):
         A List of
         List[dict] with translation
         """
+        
         try:
-            self.cache_parameters(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,max_length=max_length,add_timestamp=add_timestamp)
-            self.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
-            
-            total_progress = len(input_list_dict)
-            for index, dic in enumerate(input_list_dict):
-                progress(index / total_progress, desc="Translating..")
-                translated_text = self.translate(dic["text"], max_length=max_length)
-                dic["text"] = translated_text
+            if src_lang != tgt_lang:
+                print("translation done")
+                return input_list_dict
+                self.cache_parameters(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,max_length=max_length,add_timestamp=add_timestamp)
+                self.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
+                
+                total_progress = len(input_list_dict)
+                for index, dic in enumerate(input_list_dict):
+                    progress(index / total_progress, desc="Translating..")
+                    translated_text = self.translate(dic["text"], max_length=max_length)
+                    dic["text"] = translated_text
 
             return input_list_dict