BruceIC commited on
Commit
c161bc5
·
verified ·
1 Parent(s): 142ebfc

Update README

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -62,8 +62,8 @@ import torch
62
 
63
  classifier = pipeline(
64
  "text-classification",
65
- model="YOUR_USERNAME/emoticare", # replace with your HF repo path
66
- tokenizer="YOUR_USERNAME/emoticare",
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 = "YOUR_USERNAME/emoticare" # replace with your HF repo path
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)