DarianNLP commited on
Commit
68e5ed7
·
verified ·
1 Parent(s): 7eebb64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -52,10 +52,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
52
  import torch
53
 
54
  # Load model and tokenizer
55
- model_path = "path/to/weights_for_huggingface"
56
- tokenizer = AutoTokenizer.from_pretrained(model_path)
57
  model = AutoModelForSequenceClassification.from_pretrained(
58
- model_path,
59
  num_labels=1,
60
  problem_type="regression"
61
  )
 
52
  import torch
53
 
54
  # Load model and tokenizer
55
+
56
+ tokenizer = AutoTokenizer.from_pretrained("DarianNLP/bert_sequel_beagles")
57
  model = AutoModelForSequenceClassification.from_pretrained(
58
+ "DarianNLP/bert_sequel_beagles",
59
  num_labels=1,
60
  problem_type="regression"
61
  )