Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 (
|
| 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)
|