Spaces:
Sleeping
Sleeping
New changes
Browse files
app.py
CHANGED
|
@@ -124,9 +124,6 @@ model.eval()
|
|
| 124 |
|
| 125 |
def retrieve_top_k(query,domain='legal', model_name='nlpaueb/legal-bert-base-uncased', k=8):
|
| 126 |
# Load tokenizer and model
|
| 127 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 128 |
-
model = AutoModel.from_pretrained(model_name).to(device)
|
| 129 |
-
model.eval()
|
| 130 |
|
| 131 |
#print(f"In retrive_top_k Query:{query}")
|
| 132 |
# Tokenize and embed query using mean pooling
|
|
|
|
| 124 |
|
| 125 |
def retrieve_top_k(query,domain='legal', model_name='nlpaueb/legal-bert-base-uncased', k=8):
|
| 126 |
# Load tokenizer and model
|
|
|
|
|
|
|
|
|
|
| 127 |
|
| 128 |
#print(f"In retrive_top_k Query:{query}")
|
| 129 |
# Tokenize and embed query using mean pooling
|