type correction

utils.py

@@ -19,4 +19,4 @@ def model_predict(model, tokenizer, sentences):
         outputs = model(**inputs) # get the logits
         label = np.argmax(outputs.logits.to("cpu"))
         
-    return str(labels)
+    return int(labels)