Fixing the error facing in google colab

When using (the device) in an argument getting this error in google colab
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument index in method wrapper_CUDA__index_select)

when used .to('cpu'), it runs smoothly and gets the desired result.
Fix this issue

README.md

@@ -44,7 +44,7 @@ device = "cuda"
 
 tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
 
-model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
+model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to('cpu')
 
 def paraphrase(
     question,