Update app/main_IO.py

app/main_IO.py

@@ -73,12 +73,21 @@ def get_model_path(EMBEDDING_MODEL):
     # Try current working directory
     cwd_model_path = os.path.join(os.getcwd(), "chromadb_model")
     
-    # Check which path exists
-    for path in [local_model_path, app_model_path, cwd_model_path]:
+    # Additional: Try with sentence-transformers format
+    st_model_path = os.path.join(root_path, "models", f"sentence-transformers--{EMBEDDING_MODEL}")
+    
+    # Check which path exists AND contains model files
+    for path in [local_model_path, app_model_path, cwd_model_path, st_model_path]:
         if os.path.exists(path):
-            return path
+            # Optional: Check if it actually contains model files
+            if any(f.endswith(('.bin', '.safetensors', '.json')) for f in os.listdir(path)):
+                print(f"Found local model at: {path}")
+                return path
+            else:
+                print(f"Path exists but no model files found: {path}")
     
     # If no local model found, return None to download
+    print("No local model found, will download from Hugging Face")
     return None