lambdaofgod commited on
Commit
e2e1dca
·
1 Parent(s): f6824f1

move hardcoded model names to config

Browse files
Files changed (2) hide show
  1. model_config.py +2 -0
  2. pages/1_Retrieval_App.py +3 -2
model_config.py ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ query_encoder_model_name = "lambdaofgod/query_nbow_10_5000-15"
2
+ document_encoder_model_name = "lambdaofgod/document_nbow_10_5000-15"
pages/1_Retrieval_App.py CHANGED
@@ -7,6 +7,7 @@ import sentence_transformers
7
  import streamlit as st
8
  from findkit import feature_extractors, indexes, retrieval_pipeline
9
  from toolz import partial
 
10
 
11
 
12
  def truncate_description(description, length=50):
@@ -139,7 +140,7 @@ def app_main(
139
 
140
 
141
  app_main(
142
- query_encoder_path="lambdaofgod/query_nbow_1_2000-5",
143
- document_encoder_path="lambdaofgod/document_nbow_1_2000-5",
144
  data_path="lambdaofgod/pwc_repositories_with_dependencies",
145
  )
 
7
  import streamlit as st
8
  from findkit import feature_extractors, indexes, retrieval_pipeline
9
  from toolz import partial
10
+ import model_config
11
 
12
 
13
  def truncate_description(description, length=50):
 
140
 
141
 
142
  app_main(
143
+ query_encoder_path=model_config.query_encoder_model_name,
144
+ document_encoder_path=model_config.document_encoder_model_name,
145
  data_path="lambdaofgod/pwc_repositories_with_dependencies",
146
  )