gaonkarrs commited on
Commit
040ce40
·
1 Parent(s): bc177f1

New changes

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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