Update app.py

app.py

@@ -67,7 +67,7 @@ class VietnameseChatbot:
         ]
 
     @st.cache_data
-    def _compute_embeddings(queries):
+    def _compute_embeddings(self, _queries=None):  # Add _queries parameter with underscore
         """
         Pre-compute embeddings for conversation queries
         Cached to avoid recomputing on every run
@@ -89,15 +89,10 @@ class VietnameseChatbot:
             except Exception as e:
                 print(f"Embedding error: {e}")
                 return None
-
-        # Import these arguments to make the function self-contained
-        from transformers import AutoTokenizer, AutoModel
-        tokenizer = AutoTokenizer.from_pretrained('intfloat/multilingual-e5-small')
-        model = AutoModel.from_pretrained('intfloat/multilingual-e5-small', torch_dtype=torch.float16)
-
+    
         embeddings = []
-        for query in queries:
-            embedding = embed_single_text(query['query'], tokenizer, model)
+        for conversation in self.conversation_data:  # Use self.conversation_data instead of queries
+            embedding = embed_single_text(conversation['query'], self.tokenizer, self.model)  # Use self.tokenizer and self.model
             if embedding is not None:
                 embeddings.append(embedding)
         return np.array(embeddings)