MrVicente commited on
Commit
19b6aec
·
1 Parent(s): 7c59cdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ model.eval()
28
  def run_bart(question, censor):
29
  print(question, censor)
30
 
31
- model_input = tokenizer(question_input, truncation=True, padding=True, return_tensors="pt")
32
  generated_answers_encoded = model.generate(input_ids=model_input["input_ids"].to(device),
33
  attention_mask=model_input["attention_mask"].to(device),
34
  #bad_words_ids=bad_words_ids,
 
28
  def run_bart(question, censor):
29
  print(question, censor)
30
 
31
+ model_input = tokenizer(question, truncation=True, padding=True, return_tensors="pt")
32
  generated_answers_encoded = model.generate(input_ids=model_input["input_ids"].to(device),
33
  attention_mask=model_input["attention_mask"].to(device),
34
  #bad_words_ids=bad_words_ids,