Change script.py

script.py

@@ -22,12 +22,21 @@ def run_inference(TEST_IMAGE_PATH, pipeline_model, SUBMISSION_CSV_SAVE_PATH):
 
     multiclass_predictions = pipeline_model.predict(features_multiclass)
 
-    df_predictions = pd.DataFrame({
-        "file_name": test_images,
-        "category_id": multiclass_predictions
-    })
+    df_predictions = pd.DataFrame(columns=["file_name", "category_id"])
+
+    for i in range(len(test_images)):
+        file_name = test_images[i]
+        new_row = pd.DataFrame({
+            "file_name": file_name,
+            "category_id": multiclass_predictions[i]
+        }, index=[0])
+
+        df_predictions = pd.concat([df_predictions, new_row], ignore_index=True)
 
     df_predictions.to_csv(SUBMISSION_CSV_SAVE_PATH, index=False)
+
+    print(f"Saved predictions to: {SUBMISSION_CSV_SAVE_PATH}")
+