Update src/translate/Translate.py

src/translate/Translate.py

@@ -55,7 +55,7 @@ def paraphraseTranslateMethod(requestValue: str, model: str):
 def gemma(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-better'):
     requestValue = requestValue.replace('\n', ' ')
     prompt = f"Translate this to Romanian using a formal tone. Only return the translation: {requestValue}"
-    messages = [{"role": "user", "content": f"Translate this text to Romanian using a formal tone. Only return the translated text: {requestValue}"}]
+    messages = [{"role": "user", "content": f"Translate this text to Romanian using a formal tone, responding only with the translated text: {requestValue}"}]
     if '/' not in model:
         model = 'Gargaz/gemma-2b-romanian-better'
     # limit max_new_tokens to 150% of the requestValue  
@@ -64,11 +64,10 @@ def gemma(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-better'):
         "text-generation",
         model=model,
         device=-1,
-        max_new_tokens=max_new_tokens,          # Keep short to reduce verbosity
-        do_sample=False            # Use greedy decoding for determinism
+        max_new_tokens=max_new_tokens,   # Keep short to reduce verbosity
+        do_sample=False  # Use greedy decoding for determinism
     ) 
     output = pipe(messages, num_return_sequences=1, return_full_text=False)
     generated_text = output[0]["generated_text"]
     result = generated_text.split('\n', 1)[0].strip()
-    return result, model
-    # return output, model
+    return result, model