Update src/translate/Translate.py

src/translate/Translate.py

@@ -83,12 +83,8 @@ def gemma_direct(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-bette
     # Estimate output length (e.g., 50% longer)
     max_new_tokens = int(num_tokens * 1.5)
     max_new_tokens += max_new_tokens % 2  # ensure it's even
-    
-    # Token count estimation and safety check
-    # max_new_tokens = int(len(request_value) * 1.5)
-    # max_new_tokens += max_new_tokens % 2  # ensure it's even
-    
-    messages = [{"role": "user", "content": prompt]
+       
+    messages = [{"role": "user", "content": prompt}]
     tokenizer = AutoTokenizer.from_pretrained("Gargaz/gemma-2b-romanian-better")
     model = AutoModelForCausalLM.from_pretrained("Gargaz/gemma-2b-romanian-better").to(device)