Spaces:
Runtime error
Runtime error
disabled st.cache for the inference function, remove tensorflow from requiremensts
Browse files- app.py +1 -1
- 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
|