reshinthadithyan commited on
Commit
f947788
·
1 Parent(s): 8cd2bac

adding app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def generate_text(prompt):
23
  num_return_sequences=1
24
  )
25
 
26
- outputs = tokenizer.batch_decode(output_seq, skip_special_tokens=False)
27
  outputs = outputs[0].split("<bash>")[-1]
28
  return outputs
29
  st.set_page_config(
 
23
  num_return_sequences=1
24
  )
25
 
26
+ outputs = tokenizer.batch_decode(output_seq, skip_special_tokens=True)
27
  outputs = outputs[0].split("<bash>")[-1]
28
  return outputs
29
  st.set_page_config(