Update appStore/target.py
Browse files- appStore/target.py +3 -3
appStore/target.py
CHANGED
@@ -9,7 +9,7 @@ import numpy as np
|
|
9 |
import pandas as pd
|
10 |
import streamlit as st
|
11 |
from st_aggrid import AgGrid
|
12 |
-
from utils.tapp_classifier import
|
13 |
import logging
|
14 |
logger = logging.getLogger(__name__)
|
15 |
from utils.config import get_classifier_params
|
@@ -34,14 +34,14 @@ def app():
|
|
34 |
df = st.session_state.key0
|
35 |
|
36 |
#load Classifier
|
37 |
-
classifier =
|
38 |
st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
|
39 |
if len(df) > 100:
|
40 |
warning_msg = ": This might take sometime, please sit back and relax."
|
41 |
else:
|
42 |
warning_msg = ""
|
43 |
|
44 |
-
df =
|
45 |
threshold= params['threshold'])
|
46 |
st.dataframe(df)
|
47 |
st.session_state.key1 = df
|
|
|
9 |
import pandas as pd
|
10 |
import streamlit as st
|
11 |
from st_aggrid import AgGrid
|
12 |
+
from utils.tapp_classifier import load_tappClassifier, tapp_classification
|
13 |
import logging
|
14 |
logger = logging.getLogger(__name__)
|
15 |
from utils.config import get_classifier_params
|
|
|
34 |
df = st.session_state.key0
|
35 |
|
36 |
#load Classifier
|
37 |
+
classifier = load_tappClassifier(classifier_name=params['model_name'])
|
38 |
st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
|
39 |
if len(df) > 100:
|
40 |
warning_msg = ": This might take sometime, please sit back and relax."
|
41 |
else:
|
42 |
warning_msg = ""
|
43 |
|
44 |
+
df = tapp_classification(haystack_doc=df,
|
45 |
threshold= params['threshold'])
|
46 |
st.dataframe(df)
|
47 |
st.session_state.key1 = df
|