Spaces:
Running
Running
Update src/translate/Translate.py
Browse files
src/translate/Translate.py
CHANGED
@@ -58,4 +58,4 @@ def gemma(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-better'):
|
|
58 |
messages = [
|
59 |
{"role": "user", "content": f"Translate the following text to Romanian using a formal tone and provide only translation: {requestValue}"},
|
60 |
]
|
61 |
-
return pipe(messages)
|
|
|
58 |
messages = [
|
59 |
{"role": "user", "content": f"Translate the following text to Romanian using a formal tone and provide only translation: {requestValue}"},
|
60 |
]
|
61 |
+
return pipe(messages), model
|