MohitG012 commited on
Commit
3e0fdaa
·
verified ·
1 Parent(s): cdf9c58

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -4
src/streamlit_app.py CHANGED
@@ -23,10 +23,7 @@ from PIL import Image as PILImage
23
  from sentence_transformers import SentenceTransformer
24
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration, CLIPProcessor, CLIPModel, BitsAndBytesConfig
25
  import faiss
26
-
27
- # os.makedirs("/tmp/hf_cache", exist_ok=True)
28
- # os.makedirs("/tmp/hf_home", exist_ok=True)
29
- # os.makedirs("/tmp/sbert", exist_ok=True)
30
 
31
  # Load assets
32
  @st.cache_resource
 
23
  from sentence_transformers import SentenceTransformer
24
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration, CLIPProcessor, CLIPModel, BitsAndBytesConfig
25
  import faiss
26
+ import pickle
 
 
 
27
 
28
  # Load assets
29
  @st.cache_resource