shri07 commited on
Commit
0a98241
1 Parent(s): ce46e25

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  model_name = 'logicreasoning/LogiT5'
2
  tokenize = AutoTokenizer.from_pretrained(model_name)
3
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
1
+ import transformers
2
+ import datasets
3
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
+ from datasets import load_dataset # if loading a dataset
5
+
6
+
7
  model_name = 'logicreasoning/LogiT5'
8
  tokenize = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)