Update utils.py

utils.py

@@ -139,6 +139,7 @@ def greedy_search(input_ids: torch.Tensor,
 #Predict
 def predict(model,
             tokenizer,
+            device,
             text,
             history,
             top_p,