Antoine Louis commited on
Commit
2bb3613
1 Parent(s): e29da5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ q_encoder = SentenceTransformer(modules=[
10
  models.Pooling(word_embedding_dimension=768, pooling_mode='cls'),
11
  ])
12
  doc_embeddings = torch.load('checkpoints/doc_embeddings.pt', map_location=torch.device('cpu'))
13
- docs = pd.DataFrame(load_dataset("antoiloui/bsard", data_files="articles_fr.csv")['train'])
14
 
15
  def search(question):
16
  q_emb = q_encoder.encode(question, convert_to_tensor=True)
 
10
  models.Pooling(word_embedding_dimension=768, pooling_mode='cls'),
11
  ])
12
  doc_embeddings = torch.load('checkpoints/doc_embeddings.pt', map_location=torch.device('cpu'))
13
+ docs = pd.DataFrame(load_dataset("maastrichtlawtech/bsard", data_files="articles_fr.csv")['train'])
14
 
15
  def search(question):
16
  q_emb = q_encoder.encode(question, convert_to_tensor=True)