ppsingh commited on
Commit
de045f6
1 Parent(s): 48822ff

Update appStore/target.py

Browse files
Files changed (1) hide show
  1. 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 load_targetClassifier, tapp_classification
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 = load_targetClassifier(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 = target_classification(haystack_doc=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