Update utils.py

utils.py

@@ -137,7 +137,9 @@ def greedy_search(input_ids: torch.Tensor,
 
 ########################################
 #Predict
-def predict(text,
+def predict(model,
+            tokenizer,
+            text,
             history,
             top_p,
             temperature,