Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 import re
 from huggingface_hub import InferenceClient
@@ -7,9 +6,11 @@ import os
 import numpy as np
 from openai import OpenAI
 
+os.makedirs('./hf_cache', exist_ok=True)
+os.environ['TRANSFORMERS_CACHE'] = './hf_cache'
+os.environ['HF_HOME'] = './hf_cache'
 
-os.environ['TRANSFORMERS_CACHE'] = './model_cache'
-os.environ['HF_HOME'] = './model_cache'
+from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')