fix: type issues on the final json output

cloudbuild.yaml

@@ -6,6 +6,7 @@ steps:
     automapSubstitutions: true
 
   - name: "gcr.io/cloud-builders/docker"
+    id: "push-image"
     script: |
       docker push us-central1-docker.pkg.dev/${PROJECT_ID}/interview-ai-detector/model-prediction:latest
     automapSubstitutions: true
@@ -38,6 +39,8 @@ steps:
 
   - name: "gcr.io/google.com/cloudsdktool/cloud-sdk"
     entrypoint: "bash"
+    waitFor:
+      - "push-image"
     args:
       - "-c"
       - |

core-model-prediction/prediction.py

@@ -64,7 +64,7 @@ def process_instance(data: PredictRequest):
     return {
         "prediction_class": "AI" if secondary_model_prediction == 1 else "HUMAN",
         "details": {
-            "main_model_probability": main_model_probability,
+            "main_model_probability": str(main_model_probability),
             "final_prediction": secondary_model_prediction
         }
     }

core-model-prediction/random_forest_model.py

@@ -12,4 +12,4 @@ class RandomForestModel:
         return self.scaler.transform(np.array(secondary_model_features).astype(np.float32).reshape(1, -1))
 
     def predict(self, secondary_model_features: List[float]):
-        return self.model.predict(self.preprocess_input(secondary_model_features))[0]
+        return int(self.model.predict(self.preprocess_input(secondary_model_features))[0])