Update translator.py

translator.py

@@ -368,9 +368,16 @@ def handle_translation_request(request):
                 tokenized = tokenizer(input_text, return_tensors="pt", padding=True)
                 tokenized = {k: v.to(model.device) for k, v in tokenized.items()}
 
-                # Generate translation
                 with torch.no_grad():
-                    translated = model.generate(**tokenized)
+                    translated = model.generate(
+                        **tokenized,
+                        max_length=100,              # Reasonable output length
+                        num_beams=4,                 # Same as in training
+                        length_penalty=0.6,          # Same as in training
+                        early_stopping=True,         # Same as in training
+                        repetition_penalty=1.5,      # Add this to prevent repetition
+                        no_repeat_ngram_size=3       # Add this to prevent repetition
+                    )
 
                 # Decode the translation
                 result = tokenizer.decode(translated[0], skip_special_tokens=True)