Spaces:
Sleeping
Sleeping
Update src/translate/Translate.py
Browse files
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: {
|
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)
|