loubnabnl HF Staff commited on
Commit
71b0ec2
·
1 Parent(s): ca08216

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def generate(gen_prompt, max_tokens, temperature=0.6, seed=42):
25
  output = model.generate(input_ids=input_ids, do_sample=True, top_p=0.95, temperature=temperature, max_length=max_length)
26
  generated_text = tokenizer.decode(output.flatten())
27
  if generated_text.startswith(BOS):
28
- generated_text = detok_hypo_str[len(BOS):]
29
  return generated_text
30
 
31
  iface = gr.Interface(
 
25
  output = model.generate(input_ids=input_ids, do_sample=True, top_p=0.95, temperature=temperature, max_length=max_length)
26
  generated_text = tokenizer.decode(output.flatten())
27
  if generated_text.startswith(BOS):
28
+ generated_text = generated_text[len(BOS):]
29
  return generated_text
30
 
31
  iface = gr.Interface(