Update app.py

app.py

@@ -37,7 +37,6 @@ def load_embedding_model():
 
 model = load_embedding_model()
 
-# Inicialização do cliente Groq
 client = Groq(
     api_key=st.secrets["GROQ_API_KEY"],
 )
@@ -48,12 +47,13 @@ def load_index_and_embeddings(index_file: str, embeddings_file: str):
     embeddings = np.load(embeddings_file)
     return index, embeddings
 
+@st.cache_resource
 def search(query: str, index, embeddings: np.ndarray, chunks: list, k: int = 5):
     query_vector = model.encode([query])
     distances, indices = index.search(query_vector.astype('float32'), k)
     return [chunks[i] for i in indices[0]]
 
-@st.cache_resource
+
 def query_groq(prompt, client):
     chat_completion = client.chat.completions.create(
         messages=[