PierreHanna commited on
Commit
a8c9a19
·
verified ·
1 Parent(s): eb392be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def process(prompt, lang):
45
  embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
46
  print("Embed time : ", time.time() - a)
47
  do_normalize(embed_query)
48
- D, I = get_distance(index, embed_query, TOP)
49
  print("Search + Embed time : ", time.time() - a)
50
 
51
  #print(I)
 
45
  embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
46
  print("Embed time : ", time.time() - a)
47
  do_normalize(embed_query)
48
+ D, I = get_distance(index, embed_query, max_output)
49
  print("Search + Embed time : ", time.time() - a)
50
 
51
  #print(I)