Spaces:
Sleeping
Sleeping
tensorkelechi
commited on
Commit
•
e4c4ba8
1
Parent(s):
a646756
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def embed_data(dataset):
|
|
44 |
dataset, retrieval_type="text-image", dataset_type="huggingface"
|
45 |
)
|
46 |
embedded_data = embedder.create_embeddings(device="cpu")
|
47 |
-
return embedded_data
|
48 |
|
49 |
@stl.cache_resource
|
50 |
def init_search(_embedded_data, embedder):
|
|
|
44 |
dataset, retrieval_type="text-image", dataset_type="huggingface"
|
45 |
)
|
46 |
embedded_data = embedder.create_embeddings(device="cpu")
|
47 |
+
return embedded_data, embedder
|
48 |
|
49 |
@stl.cache_resource
|
50 |
def init_search(_embedded_data, embedder):
|