Update README
Browse files
README.md
CHANGED
|
@@ -62,8 +62,8 @@ import torch
|
|
| 62 |
|
| 63 |
classifier = pipeline(
|
| 64 |
"text-classification",
|
| 65 |
-
model="
|
| 66 |
-
tokenizer="
|
| 67 |
top_k=None, # return scores for all labels
|
| 68 |
device=0 if torch.cuda.is_available() else -1,
|
| 69 |
)
|
|
@@ -94,7 +94,7 @@ import torch
|
|
| 94 |
import torch.nn.functional as F
|
| 95 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
| 96 |
|
| 97 |
-
model_name = "
|
| 98 |
|
| 99 |
tokenizer = DistilBertTokenizer.from_pretrained(model_name)
|
| 100 |
model = DistilBertForSequenceClassification.from_pretrained(model_name)
|
|
|
|
| 62 |
|
| 63 |
classifier = pipeline(
|
| 64 |
"text-classification",
|
| 65 |
+
model="BruceIC/emoticare", # replace with your HF repo path
|
| 66 |
+
tokenizer="BruceIC/emoticare",
|
| 67 |
top_k=None, # return scores for all labels
|
| 68 |
device=0 if torch.cuda.is_available() else -1,
|
| 69 |
)
|
|
|
|
| 94 |
import torch.nn.functional as F
|
| 95 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
| 96 |
|
| 97 |
+
model_name = "BruceIC/emoticare" # replace with your HF repo path
|
| 98 |
|
| 99 |
tokenizer = DistilBertTokenizer.from_pretrained(model_name)
|
| 100 |
model = DistilBertForSequenceClassification.from_pretrained(model_name)
|