Update src/translate/Translate.py

src/translate/Translate.py

@@ -54,18 +54,18 @@ def paraphraseTranslateMethod(requestValue: str, model: str):
     return " ".join(result_value).strip(), model
 
 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}"
+    prompt = f"Translate this to Romanian using a formal tone. Only return the translation: {requestValue}"
     messages = [
-        {"role": "user", "content": f"Translate this to Romanian using a formal tone. Only return the translated text: {requestValue}"},
+        {"role": "user", "content": f"Translate this text to Romanian using a formal tone. Only return the translated text: {requestValue}"},
     ]
+    if '/' not in model:
+        model = 'Gargaz/gemma-2b-romanian-better'
     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)
+        do_sample=False            # Use greedy decoding for determinism
+    ) 
+    output = pipe(messages, num_return_sequences=1, return_full_text=False)
     return output[0]["generated_text"].strip(), model