Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -71,10 +71,10 @@ sentences = [
|
|
71 |
"dans un hôtel avec piscine à marseille"
|
72 |
]
|
73 |
inputs = tokenizer(sentences, padding=True, return_tensors='pt')
|
74 |
-
|
75 |
print([
|
76 |
[model.config.id2label[i] for i in b]
|
77 |
-
for b in
|
78 |
])
|
79 |
```
|
80 |
|
|
|
71 |
"dans un hôtel avec piscine à marseille"
|
72 |
]
|
73 |
inputs = tokenizer(sentences, padding=True, return_tensors='pt')
|
74 |
+
outputs = model(**inputs).logits
|
75 |
print([
|
76 |
[model.config.id2label[i] for i in b]
|
77 |
+
for b in outputs.argmax(dim=-1).tolist()
|
78 |
])
|
79 |
```
|
80 |
|