awacke1 commited on
Commit
34239ff
Β·
1 Parent(s): 941ae85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -103,11 +103,11 @@ def extracted_query_embeddings(queries, max_length=64):
103
  return_token_type_ids=True
104
  )
105
 
106
- query_embeddings = query_encoder.predict([np.array(query_inputs['input_ids']),
107
- np.array(query_inputs['attention_mask']),
108
- np.array(query_inputs['token_type_ids'])],
109
- batch_size=1,
110
- verbose=1)
111
  return query_embeddings
112
 
113
  #Wikipedia API:
 
103
  return_token_type_ids=True
104
  )
105
 
106
+ query_embeddings = query_encoder.predict([np.array(query_inputs['input_ids']),
107
+ np.array(query_inputs['attention_mask']),
108
+ np.array(query_inputs['token_type_ids'])],
109
+ batch_size=1,
110
+ verbose=1)
111
  return query_embeddings
112
 
113
  #Wikipedia API: