Update README.md
Browse files
README.md
CHANGED
|
@@ -59,5 +59,5 @@ input_ids = texts["input_ids"]
|
|
| 59 |
attention_mask = texts["attention_mask"]
|
| 60 |
|
| 61 |
output = model(input_ids, attention_mask)
|
| 62 |
-
class = output.argmax(dim=1)
|
| 63 |
```
|
|
|
|
| 59 |
attention_mask = texts["attention_mask"]
|
| 60 |
|
| 61 |
output = model(input_ids, attention_mask)
|
| 62 |
+
class = output.argmax(dim=1) # should be 51 (blazer)
|
| 63 |
```
|