acrowth commited on
Commit
65a3485
·
1 Parent(s): 4ffd843

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -3,6 +3,9 @@ import pandas as pd
3
  import gradio as gr
4
 
5
  df=pd.read_parquet('df.parquet')
 
 
 
6
  candidats=pd.read_parquet('candidats.parquet')
7
 
8
  document_store = InMemoryDocumentStore(use_bm25=True)
@@ -24,9 +27,8 @@ def semanticsearch(query):
24
  )
25
  results=[]
26
  for document in result['documents']:
27
- result=document.to_dict()
28
- for c in ['content_type','embedding','id']:
29
- result.pop(c)
30
  results.append(result)
31
  results=pd.DataFrame(results)
32
  return results
 
3
  import gradio as gr
4
 
5
  df=pd.read_parquet('df.parquet')
6
+ dirname='lot3'
7
+
8
+ df['fileclean']=df.file.str.replace(f'.*{dirname}/[^/]+/','').str.replace('[\(\)]','').str.replace('/[^/]+$','').str.replace('/',' ').str.replace('-',' ').str.replace(' 0+',' ')
9
  candidats=pd.read_parquet('candidats.parquet')
10
 
11
  document_store = InMemoryDocumentStore(use_bm25=True)
 
27
  )
28
  results=[]
29
  for document in result['documents']:
30
+ result=document.meta
31
+ result['score']=document.score
 
32
  results.append(result)
33
  results=pd.DataFrame(results)
34
  return results