Update app.py

app.py

@@ -44,7 +44,7 @@ def embed_data(dataset):
             dataset, retrieval_type="text-image", dataset_type="huggingface"
     )
     embedded_data = embedder.create_embeddings(device="cpu")
-    return embedded_data
+    return embedded_data. embedder
 
 @stl.cache_resource
 def init_search(_embedded_data, embedder):
@@ -59,10 +59,10 @@ def get_images_from_description(description):
 
 if dataset and stl.button("embed image dataset"):
     with stl.spinner("Initializing and creating image embeddings from dataset"):
-        embedded_data = embed_data(dataset)
+        embedded_data, embedder = embed_data(dataset)
         stl.success("Successfully embedded and created image index")
 
-if embedded_data:
+if embedded_data and embedder:
     finder = init_search(embedded_data, embedder)
     
     search_term = stl.text_input("Text description/search for image")