TiberiuCristianLeon commited on
Commit
e3e1c18
·
verified ·
1 Parent(s): 37ded96

Update src/translate/Translate.py

Browse files
Files changed (1) hide show
  1. src/translate/Translate.py +1 -1
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