Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,9 @@
+cache_dir = '/tmp/hf_cache'
+os.makedirs(cache_dir, exist_ok=True)
+os.environ['TRANSFORMERS_CACHE'] = cache_dir
+os.environ['HF_HOME'] = cache_dir
+
+from sentence_transformers import SentenceTransformer
 import streamlit as st
 from sklearn.metrics.pairwise import cosine_similarity
 import re
@@ -6,12 +12,6 @@ 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'
-
-from sentence_transformers import SentenceTransformer
-
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')