update pipeline

ml6team · Mar 6, 2024 · fd24404 · fd24404
1 parent f5347f0
commit fd24404
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/pipeline.ipynb b/src/pipeline.ipynb
@@ -336,10 +336,10 @@
    "outputs": [],
    "source": [
     "image_urls = prompts.apply(\n",
-    "    \"retrieve_images_from_faiss_index_by_prompt\",\n",
+    "    \"retrieve_from_faiss_by_prompt\",\n",
     "    arguments={\n",
-    "        \"url_mapping_path\":\"gs://soy-audio-379412-embed-datacomp/12M/id_mapping\",\n",
-    "        \"faiss_index_path\":\"gs://soy-audio-379412-embed-datacomp/12M/faiss\",\n",
+    "        \"url_mapping_path\":\"hf://datasets/fondant-ai/datacomp-small-clip/id_mapping/id_mapping\",\n",
+    "        \"faiss_index_path\":\"hf://datasets/fondant-ai/datacomp-small-clip/faiss\",\n",
     "        \"num_images\": 2\n",
     "    },\n",
     ")\n",

diff --git a/src/pipeline.py b/src/pipeline.py
@@ -26,10 +26,10 @@
 )
 
 image_urls = prompts.apply(
-    "retrieve_images_from_faiss_index_by_prompt",
+    "retrieve_from_faiss_by_prompt",
     arguments={
-        "url_mapping_path":"gs://soy-audio-379412-embed-datacomp/12M/id_mapping",
-        "faiss_index_path":"gs://soy-audio-379412-embed-datacomp/12M/faiss",
+        "url_mapping_path":"hf://datasets/fondant-ai/datacomp-small-clip/id_mapping/id_mapping",
+        "faiss_index_path":"hf://datasets/fondant-ai/datacomp-small-clip/faiss",
         "num_images": 2
     },
 )