Update src/translate/Translate.py

src/translate/Translate.py

@@ -53,9 +53,19 @@ def paraphraseTranslateMethod(requestValue: str, model: str):
 
     return " ".join(result_value).strip(), model
 
-def gemma(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-better'):  
-    pipe = pipeline("text-generation", model="Gargaz/gemma-2b-romanian-better", device = -1, max_new_tokens = 512, do_sample = True, temperature = 0.2)
+def gemma(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-better'):
+    prompt = f"Translate this to Romanian using a formal tone. Only return the translation:\n{requestValue}"
     messages = [
         {"role": "user", "content": f"Translate this to Romanian using a formal tone. Only return the translated text: {requestValue}"},
     ]
-    return pipe(messages, num_return_sequences=1, return_full_text=False), model
+    pipe = pipeline(
+        "text-generation",
+        model=model,
+        device=-1,
+        max_new_tokens=512,          # Keep short to reduce verbosity
+        do_sample=False,            # Use greedy decoding for determinism
+        temperature=0.7             # Raise slightly to avoid dull output
+    )
+    
+    output = pipe(prompt, num_return_sequences=1, return_full_text=False)
+    return output[0]["generated_text"].strip(), model