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