diff --git a/src/components/DatasetMetadataModal.tsx b/src/components/DatasetMetadataModal.tsx index 95033c4..9658992 100644 --- a/src/components/DatasetMetadataModal.tsx +++ b/src/components/DatasetMetadataModal.tsx @@ -50,8 +50,8 @@ function formatLoaderInstructions(dataset: Dataset) { if (dataset.source === 'huggingface') { return { title: 'Load from Hugging Face', - body: `Use agml.data.loadhfdataset("Project-AgML/${dataset.name}") to load this dataset from Hugging Face.`, - code: `import agml\n\nloader = agml.data.loadhfdataset("Project-AgML/${dataset.name}")`, + body: `Use agml.data.hf_loader.HuggingFaceDataLoader("Project-AgML/${dataset.name}") to load this dataset from Hugging Face.`, + code: `from agml.data.hf_loader import HuggingFaceDataLoader\n\nloader = HuggingFaceDataLoader("Project-AgML/${dataset.name}")`, }; } diff --git a/static/data/hf_datasets.json b/static/data/hf_datasets.json index f060d5f..13b9232 100644 --- a/static/data/hf_datasets.json +++ b/static/data/hf_datasets.json @@ -6,7 +6,7 @@ "location": [ "Italy" ], - "environment": "real", + "environment": "field", "crop_types": [ "grapes" ], @@ -27,9 +27,10 @@ "source": "huggingface" }, { - "name": "Project-AgML/Strawberry-DS_strawberry_detection", + "name": "Strawberry-DS_strawberry_detection", "machine_learning_task": "object_detection", "agricultural_task": "crop_maturity_monitoring", + "real_or_synthetic": "real", "location": "Egypt", "environment": "field", "crop_types": [ @@ -57,5 +58,32 @@ "parent_dataset": null, "zip_size_bytes": 293000000, "source": "huggingface" + }, + { + "name": "synthetic_cowpea_flower_detection", + "machine_learning_task": "object_detection", + "agricultural_task": "crop_detection", + "environment": "field", + "real_or_synthetic": "synthetic", + "crop_types": [ + "cowpea" + ], + "sensor_modality": "rgb", + "platform": "ground", + "input_data_format": "image_folder", + "annotation_format": "coco_json", + "num_images": 1634, + "augmented_num_images": 8170, + "classes": [ + "Flower" + ], + "stats_mean": null, + "stats_std": null, + "examples_image_url": null, + "license": null, + "citation": null, + "parent_dataset": null, + "zip_size_bytes": 246700000, + "source": "huggingface" } ] \ No newline at end of file