Spaces:
Running
Running
Update src/translate/Translate.py
Browse files
src/translate/Translate.py
CHANGED
|
@@ -99,7 +99,7 @@ def gemma_direct(requestValue: str, model: str = 'Gargaz/gemma-2b-romanian-bette
|
|
| 99 |
|
| 100 |
outputs = model.generate(**inputs, max_new_tokens=max_new_tokens)
|
| 101 |
response = tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
|
| 102 |
-
return response
|
| 103 |
except Exception as error:
|
| 104 |
return error
|
| 105 |
|
|
|
|
| 99 |
|
| 100 |
outputs = model.generate(**inputs, max_new_tokens=max_new_tokens)
|
| 101 |
response = tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
|
| 102 |
+
return response.strip()
|
| 103 |
except Exception as error:
|
| 104 |
return error
|
| 105 |
|