Update README.md
Browse files
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
|