Update app.py

app.py

@@ -24,14 +24,14 @@ def create_faiss_index(isco_taxonomy, model_name='BAAI/bge-m3'):
     dimension = embeddings.shape[1]
     index = faiss.IndexFlatL2(dimension)
     index.add(embeddings)
-    faiss.write_index(index, 'isco_taxonomy.index')
+    faiss.write_index(index, '/data/isco_taxonomy.index')
     with open('isco_taxonomy_mapping.json', 'w') as f:
         json.dump({i: entry for i, entry in enumerate(isco_taxonomy)}, f)
 
 # Define a function to retrieve and rerank using FAISS
 def retrieve_and_rerank_faiss(job_duties, model_name="BAAI/bge-m3", top_k=4):
     # Check if isco_taxonomy.index exists, if not, create it with create_faiss_index
-    if not os.path.exists("isco_taxonomy.index"):
+    if not os.path.exists("/data/isco_taxonomy.index"):
         isco_taxonomy = load_isco_taxonomy('isco_taxonomy.jsonl')
         create_faiss_index(isco_taxonomy)
     index = faiss.read_index("isco_taxonomy.index")
@@ -47,10 +47,6 @@ def retrieve_and_rerank_faiss(job_duties, model_name="BAAI/bge-m3", top_k=4):
     ]
     return results
 
-# Load data and create index (should be done once and then commented out or moved to a setup script)
-# isco_taxonomy = load_isco_taxonomy('isco_taxonomy.jsonl')
-# create_faiss_index(isco_taxonomy)
-
 # Gradio Interface
 def gradio_interface(job_duties):
     results = retrieve_and_rerank_faiss(job_duties)