update

app.py

@@ -115,10 +115,8 @@ class TrainStudent(nn.Module):
         mse = (emb_student - emb_teacher).pow(2).mean()
         return mse
     
-chunk_data = generate_chunk_data(["AI","moon","brain"])
 student_model=torch.load("myTextEmbeddingStudent.pt",map_location='cpu').student_model.eval()
-# create the embedding vector database
-chunk_emb = generate_chunk_emb(student_model, chunk_data)
+
 
 #new_chunk_data = []
 #new_chunk_emb = tensor([])
@@ -133,6 +131,9 @@ def search(input, user_concepts):
         new_chunk_emb = generate_chunk_emb(student_model, new_chunk_data)    
         result = search_document(input, new_chunk_data, new_chunk_emb, student_model)
     else:
+        chunk_data = generate_chunk_data(["machine learning","moon","brain"])
+        # create the embedding vector database
+        chunk_emb = generate_chunk_emb(student_model, chunk_data)        
         result = search_document(input, chunk_data, chunk_emb, student_model)
 
     return " ".join(result)