TiberiuCristianLeon commited on
Commit
ad83d27
·
verified ·
1 Parent(s): 5829d30

Update src/translate/Translate.py

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