imported torch

utils.py

@@ -1,3 +1,5 @@
+import torch
+
 def model_predict(model, tokenizer, sentences):
     """
     Predict the labels of the sentences using the model and tokenizer