Update helper.py

helper.py

@@ -229,36 +229,3 @@ def get_images_with_bounding_boxes_from_s3(bucket_name, file_paths, bounding_box
         st.image(img, caption=file_path, use_column_width=True)
 
 
-def main():
-    print('Begin Main')
-    dataset_name = "WayveScenes"
-    query = "black car"
-    limit = 10
-    offset = 0
-    scoring_func = "cosine"
-    search_in_images = True
-    search_in_small_objects = True
-    dataset = load_dataset(f"quasara-io/{dataset_name}")
-    print('loaded dataset')
-    dataset_keys = dataset.keys()
-    random_sample_size = 1000
-    main_df, split_df = parallel_load_and_combine(dataset_keys, dataset, n_rows=random_sample_size)
-    #Now we get the coordinates and the stuff
-    print('processed datasets')
-    if search_in_small_objects:
-        results = batch_search(query, split_df)
-        print(results)
-        top_k_paths = get_file_paths(split_df,results) 
-        top_k_cordinates = get_cordinates(split_df, results)
-        print(top_k_paths)
-        print(top_k_cordinates)
-        return top_k_paths, top_k_cordinates
-    else:
-        results = search(query, main_df, limit, offset, scoring_func, search_in_images)
-        top_k_paths = get_file_paths(main_df,results) 
-        print(top_k_paths)
-        return top_k_paths
-
-
-if __name__ == "__main__":
-    main()