Add to(device) to "input_ids"

If the model is taken to the GPU but not the tokenized "input ids" then the following runtime error will be raised:
`RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu`
Therefore, .input_ids should be modified to .input_ids.to(device).

Ref:
https://stackoverflow.com/a/75831568/10216028

README.md

@@ -65,7 +65,7 @@ def paraphrase(
         return_tensors="pt", padding="longest",
         max_length=max_length,
         truncation=True,
-    ).input_ids
+    ).input_ids.to(device)
     
     outputs = model.generate(
         input_ids, temperature=temperature, repetition_penalty=repetition_penalty,