Spaces:
Runtime error
Runtime error
Commit
·
e2e1dca
1
Parent(s):
f6824f1
move hardcoded model names to config
Browse files- model_config.py +2 -0
- 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=
|
143 |
-
document_encoder_path=
|
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 |
)
|