Update README.md

README.md

@@ -8,12 +8,12 @@ pip install transformers[torch]
 ```
 
 ```python
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import time
 
 tokenizer = AutoTokenizer.from_pretrained("AquilaX-AI/DB-Summarizer")
-model = AutoModelForSeq2SeqLM.from_pretrained("AquilaX-AI/DB-Summarizer")
+model = AutoModelForCausalLM.from_pretrained("AquilaX-AI/DB-Summarizer")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -28,7 +28,7 @@ inputs = tokenizer(summ_inp, return_tensors="pt")
 model.to(device)
 inputs = inputs.to(device)
 outputs = model.generate(**inputs, max_length=526)
-answer = summ_tokenizer.decode(outputs[0], skip_special_tokens=True)
+answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
 print(answer)
 
 end = time.time()