joe32140 commited on
Commit
916e531
·
verified ·
1 Parent(s): 751d3a1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
33
 
34
  # Load the model and tokenizer
35
  model_name = "joe32140/flan-t5-large-claim-category"
36
- tokenizer = T5Tokenizer.from_pretrained(model_name)
37
  model = T5ForConditionalGeneration.from_pretrained(model_name)
38
 
39
  # Prepare the input
 
33
 
34
  # Load the model and tokenizer
35
  model_name = "joe32140/flan-t5-large-claim-category"
36
+ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
37
  model = T5ForConditionalGeneration.from_pretrained(model_name)
38
 
39
  # Prepare the input