TiberiuCristianLeon commited on
Commit
d550028
·
verified ·
1 Parent(s): b48c77c

Update src/translate/Translate.py

Browse files
Files changed (1) hide show
  1. src/translate/Translate.py +1 -1
src/translate/Translate.py CHANGED
@@ -76,7 +76,7 @@ def gemma_direct(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-bette
76
  # Load model directly
77
  model = model if '/' in model else 'Gargaz/gemma-2b-romanian-better'
78
  # limit max_new_tokens to 150% of the requestValue
79
- prompt = f"Translate this text to Romanian: {request_value}"
80
 
81
  input_ids = tokenizer.encode(request_value, add_special_tokens=True)
82
  num_tokens = len(input_ids)
 
76
  # Load model directly
77
  model = model if '/' in model else 'Gargaz/gemma-2b-romanian-better'
78
  # limit max_new_tokens to 150% of the requestValue
79
+ prompt = f"Translate this text to Romanian: {requestValue}"
80
 
81
  input_ids = tokenizer.encode(request_value, add_special_tokens=True)
82
  num_tokens = len(input_ids)