vpelloin commited on
Commit
bde5c1b
1 Parent(s): 31435f9

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- outptus = model(**inputs).logits
75
  print([
76
  [model.config.id2label[i] for i in b]
77
- for b in outptus.argmax(dim=-1).tolist()
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