Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ output_ids = model.generate(**input_ids, num_beams=1, do_sample=False, max_new_t
|
|
| 58 |
|
| 59 |
# output should be either yer or no
|
| 60 |
answer = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 61 |
-
|
| 62 |
```
|
| 63 |
|
| 64 |
|
|
|
|
| 58 |
|
| 59 |
# output should be either yer or no
|
| 60 |
answer = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 61 |
+
prediction = "yes" in answer.lower()
|
| 62 |
```
|
| 63 |
|
| 64 |
|