Add move data util nb and update config

Update model prefix in infer_kpi nb Co-authored-by: Karanraj Chauhan <[email protected]> Signed-off-by: Shreyanand <[email protected]>
os-climate · Mar 18, 2022 · cbe7686 · cbe7686
1 parent b7711a2
commit cbe7686
Show file tree

Hide file tree

Showing 4 changed files with 453 additions and 96 deletions.
diff --git a/notebooks/demo2/config.py b/notebooks/demo2/config.py
@@ -22,16 +22,16 @@
 BASE_CURATION_FOLDER = DATA_FOLDER / "curation"
 BASE_INFER_KPI_FOLDER = DATA_FOLDER / "infer_KPI"
 
-CHECKPOINT_S3_PREFIX = "corpdata/saved_models"
-DATA_S3_PREFIX = "corpdata/ESG/pipeline_run/samples_10"
+CHECKPOINT_S3_PREFIX = "aicoe-osc-demo/saved_models"
+DATA_S3_PREFIX = "aicoe-osc-demo/pipeline_run/samples_4"
 BASE_PDF_S3_PREFIX = f"{DATA_S3_PREFIX}/pdfs"
 BASE_ANNOTATION_S3_PREFIX = f"{DATA_S3_PREFIX}/annotations"
 BASE_EXTRACTION_S3_PREFIX = f"{DATA_S3_PREFIX}/extraction"
 BASE_CURATION_S3_PREFIX = f"{DATA_S3_PREFIX}/curation"
 BASE_INFER_RELEVANCE_S3_PREFIX = f"{DATA_S3_PREFIX}/infer_relevance"
 BASE_INFER_KPI_S3_PREFIX = f"{DATA_S3_PREFIX}/infer_KPI"
 
-BASE_INFER_KPI_TABLE_S3_PREFIX = "corpdata/ESG/KPI_table"
+BASE_INFER_KPI_TABLE_S3_PREFIX = "aicoe-osc-demo/KPI_table"
 
 ckpt = "icdar_19b2_v2.pth"
 config_file = "cascade_mask_rcnn_hrnetv2p_w32_20e_v2.py"