tskolm commited on
Commit
d366ba4
·
1 Parent(s): bc652a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def generate(tokenizer, model, text, features):
13
  max_length=features['max_length'], top_p=features['top_p'], temperature=features['t'] / 100.0, num_return_sequences=features['num'],
14
  )
15
  for i, sample_output in enumerate(sample_outputs):
16
- decoded = tokenizer.decode(sample_output, skip_special_tokens=True).replace('//', '/').split(text)[1]
17
  st.write(decoded)
18
 
19
 
 
13
  max_length=features['max_length'], top_p=features['top_p'], temperature=features['t'] / 100.0, num_return_sequences=features['num'],
14
  )
15
  for i, sample_output in enumerate(sample_outputs):
16
+ decoded = tokenizer.decode(sample_output, skip_special_tokens=True).replace('\\\\', '\\').split(text)[1]
17
  st.write(decoded)
18
 
19