Upload folder using huggingface_hub

script.py

@@ -102,11 +102,12 @@ def make_submission(test_metadata, model_path, model_name, output_csv_path="./su
     model = PytorchWorker()
     data = LoadImagesAndLabels(test_metadata, VALID_AUG, mode='test')
     predictions = []
+
     for image in data:
         output = model.predict_image(image)
         string_label_dup = LABELS.get(str(np.argmax(output)), 'Acanthophis antarcticus')
         prediction_class = ORIGINAL_LABELS.get(string_label_dup, 1)
-        predictions.append(np.argmax(prediction_class))
+        predictions.append(prediction_class)
     print(predictions)
 
     test_metadata["class_id"] = predictions
@@ -128,6 +129,9 @@ if __name__ == "__main__":
     metadata_file_path = "./SnakeCLEF2024_TestMetadata.csv"
     test_metadata = pd.read_csv(metadata_file_path)
 
+    # test_metadata = pd.DataFrame()
+    # test_metadata['filename'] = ['../sample.png', '../sample copy.png', '../sample copy 2.png']
+    # test_metadata['observation_id'] = [1, 2, 3]
     make_submission(
         test_metadata=test_metadata,
         model_path=MODEL_PATH,