logicreasoning
/

LogiT5

Text2Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

shri07 commited on Nov 2, 2023

Commit

6d46d58

•

1 Parent(s): b3eea3c

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -1,4 +1,3 @@
-```
 import transformers
 import datasets
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -13,4 +12,4 @@ input_text = '' #your input text here must be a string
 input = tokenize(input_text, return_tensors='pt', padding=True).to(device)
 model = model.to(device)
 output = model.generate(*input, max_length=1024)
-prediction = tokenize.decode(output[0],skip_special_tokens=True)```

 import transformers
 import datasets
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 input = tokenize(input_text, return_tensors='pt', padding=True).to(device)
 model = model.to(device)
 output = model.generate(*input, max_length=1024)
+prediction = tokenize.decode(output[0],skip_special_tokens=True)