Update translation.py

translation.py

@@ -77,8 +77,11 @@ class CombinedModel:
             if not input_ids or input_ids.size(0) == 0:
                 return torch.tensor([])
             inputs = self.default_tokenizer.batch_decode(input_ids, skip_special_tokens=True)
-            translated = [combined_translate(text, self.source_lang, self.target_lang, self.default_tokenizer, self.default_model) for text in inputs]
-            return torch.tensor([self.default_tokenizer.encode(t, return_tensors="pt", padding=True, truncation=True, max_length=500)[0] for t in translated])
+            translated_texts = [combined_translate(text, self.source_lang, self.target_lang, self.default_tokenizer, self.default_model) for text in inputs]
+            encoded_outputs = [self.default_tokenizer.encode(t, return_tensors="pt", padding=True, truncation=True, max_length=500)[0] for t in translated_texts]
+            if not encoded_outputs:
+                return torch.tensor([])
+            return torch.stack(encoded_outputs)  # Stack tensors to ensure proper shape
         except Exception as e:
             st.error(f"Generation error in CombinedModel: {e}")
             return torch.tensor([])
@@ -129,8 +132,14 @@ def translate(text, source_lang, target_lang):
             inputs = {k: v[0].unsqueeze(0) for k, v in inputs.items()}
         with torch.no_grad():
             translated_ids = model.generate(**inputs, max_length=1000 if target_lang in ["hi", "zh", "ja"] else 500, num_beams=4, early_stopping=True)
-        result = tokenizer.decode(translated_ids[0], skip_special_tokens=True)
-        return result if result.strip() else text
+        result = tokenizer.decode(translated_ids[0], skip_special_tokens=True) if translated_ids.size(0) > 0 else None
+        if result and result.strip():
+            return result
+        else:
+            message = "This translation is not possible at this moment. Please try another language."
+            st.warning(message)
+            return f"{text} (Note: {message})"
     except Exception as e:
         st.error(f"Translation failed: {e}")
-        return text
+        message = "This translation is not possible at this moment. Please try another language."
+        return f"{text} (Note: {message})"