Add to(device) to "input_ids"

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,