tensorkelechi commited on
Commit
72c69ab
·
verified ·
1 Parent(s): 7f88749

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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