cahya commited on
Commit
822e91a
1 Parent(s): b7ac871

disabled st.cache for the inference function, remove tensorflow from requiremensts

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def get_wit_index():
18
  wit_index = WitIndex(wit_index_path, model_name, wit_dataset_path, gpu=False)
19
  return wit_index
20
 
21
- @st.cache(suppress_st_warning=True)
22
  def process(text: str, top_k: int = 10):
23
  # st.write("Cache miss: process")
24
  distance, index, image_info = wit_index.search(text, top_k=top_k)
 
18
  wit_index = WitIndex(wit_index_path, model_name, wit_dataset_path, gpu=False)
19
  return wit_index
20
 
21
+ # @st.cache(suppress_st_warning=True)
22
  def process(text: str, top_k: int = 10):
23
  # st.write("Cache miss: process")
24
  distance, index, image_info = wit_index.search(text, top_k=top_k)
requirements.txt CHANGED
@@ -1,6 +1,5 @@
1
  numpy
2
  torch
3
- tensorflow
4
  transformers
5
  sentence-transformers
6
  datasets
 
1
  numpy
2
  torch
 
3
  transformers
4
  sentence-transformers
5
  datasets