danieldux commited on
Commit
c6d244b
1 Parent(s): 2126a12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,7 +42,7 @@ def create_faiss_index(isco_taxonomy, model_name="BAAI/bge-m3"):
42
 
43
 
44
  # Define a function to retrieve and rerank using FAISS
45
- def retrieve_and_rerank_faiss(job, model_name="BAAI/bge-m3", top_k=4):
46
  # Check if isco_taxonomy.index exists, if not, create it with create_faiss_index
47
  if not os.path.exists("/data/isco_taxonomy.index"):
48
  isco_taxonomy = load_isco_taxonomy("isco_taxonomy.jsonl")
@@ -92,7 +92,7 @@ with gr.Blocks() as demo:
92
  datatype="str",
93
  label="Results",
94
  headers=[
95
- "Score",
96
  "ISCO code",
97
  "ISCO label",
98
  "ESCO label",
 
42
 
43
 
44
  # Define a function to retrieve and rerank using FAISS
45
+ def retrieve_and_rerank_faiss(job, model_name="BAAI/bge-m3", top_k=8):
46
  # Check if isco_taxonomy.index exists, if not, create it with create_faiss_index
47
  if not os.path.exists("/data/isco_taxonomy.index"):
48
  isco_taxonomy = load_isco_taxonomy("isco_taxonomy.jsonl")
 
92
  datatype="str",
93
  label="Results",
94
  headers=[
95
+ "Distance",
96
  "ISCO code",
97
  "ISCO label",
98
  "ESCO label",