Update src/translate/Translate.py

src/translate/Translate.py

@@ -56,6 +56,6 @@ def paraphraseTranslateMethod(requestValue: str, model: str):
 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)
     messages = [
-        {"role": "user", "content": f"Translate the following text to Romanian using a formal tone and provide only translation: {requestValue}"},
+        {"role": "user", "content": f"Translate this to Romanian using a formal tone. Only return the translated text: {requestValue}"},
     ]
-    return pipe(messages, num_return_sequences=2, return_full_text=False), model
+    return pipe(messages, num_return_sequences=1, return_full_text=False), model