shri07 commited on
Commit
0209cbe
·
1 Parent(s): 4c0f5fa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -9
README.md CHANGED
@@ -2,12 +2,3 @@
2
  license: apache-2.0
3
  ---
4
 
5
- model_name = 'logicreasoning/LogiT5'
6
- tokenize = AutoTokenizer.from_pretrained(model_name)
7
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
8
- device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
9
- input_text = '' #your input text here
10
- input = tokenize(input, return_tensors='pt', padding=True).to(device)
11
- model = model.to(device)
12
- output = model.generate(*input, max_length=1024)
13
- prediction = tokenize.decode(output[0],skip_special_tokens=True)
 
2
  license: apache-2.0
3
  ---
4