Fix image path

utils/embedding_generation.py

@@ -8,7 +8,7 @@ def get_info(catalog, column):
     text_descriptions = []
     
     for index, row in catalog.iterrows():
-        path = "/content/drive/MyDrive/images/" + str(row["Id"]) + ".jpg"
+        path = "/home/user/app/images" + str(row["Id"]) + ".jpg"
         image_paths.append(path)
         text_descriptions.append(row[column])
     

utils/refine_metadata.py

@@ -47,7 +47,7 @@ def refine_metadata(catalog, column):
         
         # Ensure the image ID is converted to a string
         #image_path = "/content/drive/MyDrive/images/" + str(row["Id"]) + ".jpg"
-        image_path = "/images/" + str(row["Id"]) + ".jpg"
+        image_path = "/home/user/app/images/" + str(row["Id"]) + ".jpg"
         
         # Generate the image description using LLaVA
         refined_metadata = refine_metadata(metadata, image_path)

utils/vector_database.py

@@ -67,7 +67,7 @@ def search_in_milvus(collection, search_field, query_embedding, top_k=6):
 
         # Load the image (you can use PIL to load the image)
         #image_path = "/content/drive/MyDrive/images/" + str(image_id) + ".jpg"
-        image_path = "/images/" + str(image_id) + ".jpg"
+        image_path = "/home/user/app/images/" + str(image_id) + ".jpg"
         image = Image.open(image_path)
 
         # Append the image, metadata, and score to the search results