Antoine Louis
commited on
Commit
•
2bb3613
1
Parent(s):
e29da5d
Update app.py
Browse files
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("
|
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)
|