SkyAsl commited on
Commit
703f4b0
·
verified ·
1 Parent(s): 0c9870f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -19
README.md CHANGED
@@ -16,25 +16,6 @@ pipeline_tag: text-classification
16
  # Model Card for Model ID
17
 
18
 
19
- ## Model Details
20
-
21
-
22
- ## How to Get Started with the Model
23
-
24
- comment = ""
25
- input = tokenizer(comment, return_tensors="pt", padding =True, truncation = True)
26
- outputs = model(**{k: v.to("cuda") for k, v in input.items()})
27
- logits = outputs.logits.detach().cpu().numpy()
28
-
29
- probs = np.exp(logits) / np.exp(logits).sum(axis=-1, keepdims=True)
30
-
31
- preds = np.argmax(probs, axis=1)
32
-
33
- for text, pred, p in zip(com, preds, probs):
34
- label = id2label[int(pred)]
35
- confidence = np.round(p[int(pred)], 3)
36
- print(f"Text: {text}\nPredicted: {label} (Confidence: {confidence})\n")
37
-
38
  ## Training Details
39
 
40
  ### Training Data
 
16
  # Model Card for Model ID
17
 
18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  ## Training Details
20
 
21
  ### Training Data