shainaraza commited on
Commit
c156871
1 Parent(s): 8beec99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def get_debiased_sequence(prompt):
60
  do_sample=True,
61
  top_k=10,
62
  num_return_sequences=1,
63
- eos_token_id=tokenizer.eos_token_id,
64
  max_length=len(prompt.split(" ")) + len(input_text.split(" ")) + 100, # Increased max_length
65
  )
66
  res = sequences[0]['generated_text']
 
60
  do_sample=True,
61
  top_k=10,
62
  num_return_sequences=1,
63
+ eos_token_id=debias_tokenizer.eos_token_id,
64
  max_length=len(prompt.split(" ")) + len(input_text.split(" ")) + 100, # Increased max_length
65
  )
66
  res = sequences[0]['generated_text']