Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def embed_data(dataset):
|
|
47 |
return embedded_data
|
48 |
|
49 |
@stl.cache_resource
|
50 |
-
def init_search(_embedded_data):
|
51 |
text_search = ripple.TextSearch(_embedded_data, embedder.embed_model)
|
52 |
stl.success("Initialized text search class")
|
53 |
return text_search
|
@@ -63,7 +63,7 @@ if dataset and stl.button("embed image dataset"):
|
|
63 |
stl.success("Successfully embedded and created image index")
|
64 |
|
65 |
if embedded_data:
|
66 |
-
finder = init_search(embedded_data)
|
67 |
|
68 |
search_term = stl.text_input("Text description/search for image")
|
69 |
|
|
|
47 |
return embedded_data
|
48 |
|
49 |
@stl.cache_resource
|
50 |
+
def init_search(_embedded_data, embedder):
|
51 |
text_search = ripple.TextSearch(_embedded_data, embedder.embed_model)
|
52 |
stl.success("Initialized text search class")
|
53 |
return text_search
|
|
|
63 |
stl.success("Successfully embedded and created image index")
|
64 |
|
65 |
if embedded_data:
|
66 |
+
finder = init_search(embedded_data, embedder)
|
67 |
|
68 |
search_term = stl.text_input("Text description/search for image")
|
69 |
|