julian-schelb commited on
Commit
f80a6d9
1 Parent(s): eaf0c14

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- model_tuned = RobertaForTokenClassification.from_pretrained("./results/checkpoint-final/")
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 = model_tuned(**inputs).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