rgp230 commited on
Commit
232133b
·
1 Parent(s): 14f0dd1

feat(QuestionModel): Complete for deploy

Browse files
Files changed (2) hide show
  1. src/__init__.py +0 -0
  2. src/streamlit_app.py +3 -3
src/__init__.py ADDED
File without changes
src/streamlit_app.py CHANGED
@@ -93,9 +93,9 @@ if __name__=='__main__':
93
  user_input=ui.load_streamlit_ui()
94
  LLM_Selection=ModelSelection(user_input)
95
  if user_input["GENAI_API_KEY"]:llm=LLM_Selection.setup_llm_model()
96
- loaded_tokenizer = DistilBertTokenizerFast.from_pretrained('./train_bert/topic_classifier_model')
97
- loaded_model = TFDistilBertForSequenceClassification.from_pretrained('./train_bert/topic_classifier_model')
98
- df_keys=pd.read_csv('./train_bert/training_data/Keyword_Patterns.csv')
99
 
100
  if not user_input:
101
  st.error("Error: Failed to load user input from the UI.")
 
93
  user_input=ui.load_streamlit_ui()
94
  LLM_Selection=ModelSelection(user_input)
95
  if user_input["GENAI_API_KEY"]:llm=LLM_Selection.setup_llm_model()
96
+ loaded_tokenizer = DistilBertTokenizerFast.from_pretrained('src/train_bert/topic_classifier_model')
97
+ loaded_model = TFDistilBertForSequenceClassification.from_pretrained('src/train_bert/topic_classifier_model')
98
+ df_keys=pd.read_csv('src/train_bert/training_data/Keyword_Patterns.csv')
99
 
100
  if not user_input:
101
  st.error("Error: Failed to load user input from the UI.")