Update translation.py

translation.py

@@ -31,7 +31,7 @@ def _load_all_models():
 
 all_models = _load_all_models()
 
-# Define combined_translate outside load_model
+# Define combined_translate outside load_model with explicit parameters
 def combined_translate(text, source_lang, target_lang, default_tokenizer, default_model):
     with torch.no_grad():
         if source_lang != "en":
@@ -53,7 +53,10 @@ class CombinedModel:
         self.default_model = default_model
 
     def generate(self, **kwargs):
-        return torch.tensor([combined_translate(tokenizer.decode(x, skip_special_tokens=True), self.source_lang, self.target_lang, self.default_tokenizer, self.default_model) for x in kwargs['input_ids']])
+        input_ids = kwargs.get('input_ids')
+        if not input_ids:
+            return torch.tensor([])
+        return torch.tensor([combined_translate(tokenizer.decode(x, skip_special_tokens=True), self.source_lang, self.target_lang, self.default_tokenizer, self.default_model) for x in input_ids])
 
 # Function to load appropriate translation model with optimized caching
 @st.cache_resource