Spaces:
Build error
Build error
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 |
|