RoAr777 commited on
Commit
deb2667
·
1 Parent(s): 765a5b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def chatbot_model(m,history):
27
 
28
  # Decode the output
29
  decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
30
- if (decode_output=='NO'):
31
  yield 'NO'
32
  input_ids = tokenizer.encode(prompt2.format(m) + tokenizer.eos_token, return_tensors='pt')
33
  outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
 
27
 
28
  # Decode the output
29
  decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
30
+ if (decoded_output=='NO'):
31
  yield 'NO'
32
  input_ids = tokenizer.encode(prompt2.format(m) + tokenizer.eos_token, return_tensors='pt')
33
  outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)