Update app.py

app.py

@@ -16,7 +16,11 @@ CHROMA_PATH = "chroma"
 DATA_PATH = ""  # Update this to your actual data path
 
 def main():
-
+    if not os.path.exists("model_cache"):
+        os.makedirs("model_cache")
+    if not os.path.exists("chroma"):
+        os.makedirs("chroma")
+    
     generate_data_store()
 
 def generate_data_store():