julian-schelb
commited on
Commit
•
f80a6d9
1
Parent(s):
eaf0c14
Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ This model is limited by its training dataset of entity-annotated news articles
|
|
30 |
|
31 |
```python
|
32 |
|
33 |
-
|
34 |
|
35 |
text = "Für Richard Phillips Feynman war es immer wichtig in New York, die unanschaulichen Gesetzmäßigkeiten der Quantenphysik Laien und Studenten nahezubringen und verständlich zu machen."
|
36 |
|
@@ -40,7 +40,7 @@ inputs = tokenizer(
|
|
40 |
)
|
41 |
|
42 |
with torch.no_grad():
|
43 |
-
logits =
|
44 |
|
45 |
predicted_token_class_ids = logits.argmax(-1)
|
46 |
|
|
|
30 |
|
31 |
```python
|
32 |
|
33 |
+
model = RobertaForTokenClassification.from_pretrained("julian-schelb/roberta-ner-multilingual/")
|
34 |
|
35 |
text = "Für Richard Phillips Feynman war es immer wichtig in New York, die unanschaulichen Gesetzmäßigkeiten der Quantenphysik Laien und Studenten nahezubringen und verständlich zu machen."
|
36 |
|
|
|
40 |
)
|
41 |
|
42 |
with torch.no_grad():
|
43 |
+
logits = model(**inputs).logits
|
44 |
|
45 |
predicted_token_class_ids = logits.argmax(-1)
|
46 |
|