aveluth commited on
Commit
bf630f0
·
1 Parent(s): 5987677

updated readme

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -35,7 +35,7 @@ import torch
35
  model = AutoModelForSequenceClassification.from_pretrained("aveluth/author_regulatory_focus_classifier")
36
  tokenizer = AutoTokenizer.from_pretrained("aveluth/author_regulatory_focus_classifier")
37
 
38
- text = ""
39
  inputs = tokenizer(text, return_tensors="pt")
40
  outputs = model(**inputs)
41
  predicted_class = torch.argmax(outputs.logits).item()
@@ -43,8 +43,6 @@ predicted_class = torch.argmax(outputs.logits).item()
43
  print("Predicted class:", "prevention" if predicted_class == 0 else "promotion")
44
  ```
45
 
46
- > Make sure to replace `"your-username/..."` with the correct model path.
47
-
48
  ## Labels
49
 
50
  | Class | Description |
@@ -83,4 +81,4 @@ If you use this model in your research, please cite:
83
  number={1},
84
  year={2023}
85
  }
86
- ```
 
35
  model = AutoModelForSequenceClassification.from_pretrained("aveluth/author_regulatory_focus_classifier")
36
  tokenizer = AutoTokenizer.from_pretrained("aveluth/author_regulatory_focus_classifier")
37
 
38
+ text = "Wir müssen sicherstellen, dass keine Fehler passieren. Sicherheit hat höchste Priorität."
39
  inputs = tokenizer(text, return_tensors="pt")
40
  outputs = model(**inputs)
41
  predicted_class = torch.argmax(outputs.logits).item()
 
43
  print("Predicted class:", "prevention" if predicted_class == 0 else "promotion")
44
  ```
45
 
 
 
46
  ## Labels
47
 
48
  | Class | Description |
 
81
  number={1},
82
  year={2023}
83
  }
84
+ ```