umarbutler commited on
Commit
1afde0c
1 Parent(s): 5eb8531

Fixing typos.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -83,10 +83,10 @@ from transformers import AutoModel, AutoTokenizer
83
  BATCH_SIZE = 8
84
 
85
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
86
- model = AutoModel.from_pretrained('model').to(device)
87
  model = model.to_bettertransformer() # Optional: convert the model into a BetterTransformer
88
  # to speed it up.
89
- tokeniser = AutoTokenizer.from_pretrained('model')
90
 
91
  texts = [
92
  'The Parliament shall, subject to this Constitution,\
 
83
  BATCH_SIZE = 8
84
 
85
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
86
+ model = AutoModel.from_pretrained('umarbutler/emubert').to(device)
87
  model = model.to_bettertransformer() # Optional: convert the model into a BetterTransformer
88
  # to speed it up.
89
+ tokeniser = AutoTokenizer.from_pretrained('umarbutler/emubert')
90
 
91
  texts = [
92
  'The Parliament shall, subject to this Constitution,\