fix path issue
This commit is contained in:
parent
7c39778ea1
commit
5633019411
4
.env_template
Normal file
4
.env_template
Normal file
@ -0,0 +1,4 @@
|
||||
RECIPE_PATH=AMUseBotFront/ai_talks/AMUseBotBackend/recipe/
|
||||
DIALOG_PATH=AMUseBotFront/ai_talks/AMUseBotBackend/dialog/
|
||||
INTENT_DICT_PATH=ai_talks/AMUseBotBackend/utils/intent_dict.json
|
||||
MODEL_IDENTIFIER_PATH=ai_talks/AMUseBotBackend/models/NLU/roberta-base-cookdial.txt
|
@ -10,6 +10,8 @@ from PIL import Image
|
||||
from src.utils.conversation import get_user_input, show_chat_buttons, show_conversation
|
||||
from src.utils.lang import en
|
||||
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
@ -37,6 +39,11 @@ if __name__ == '__main__':
|
||||
with open(css_file) as f:
|
||||
st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
|
||||
|
||||
load_dotenv()
|
||||
|
||||
DIALOG_PATH = os.getenv('DIALOG_PATH')
|
||||
RECIPE_PATH = os.getenv('RECIPE_PATH')
|
||||
|
||||
# Storing The Context
|
||||
if "locale" not in st.session_state:
|
||||
st.session_state.locale = en
|
||||
@ -57,7 +64,7 @@ if __name__ == '__main__':
|
||||
if "total_tokens" not in st.session_state:
|
||||
st.session_state.total_tokens = []
|
||||
if "dst" not in st.session_state:
|
||||
st.session_state.dst = DST(recipe_path="AMUseBotFront/ai_talks/AMUseBotBackend/recipe/", dialog_path="AMUseBotFront/ai_talks/AMUseBotBackend/dialog/")
|
||||
st.session_state.dst = DST(recipe_path=RECIPE_PATH, dialog_path=DIALOG_PATH)
|
||||
if "dp" not in st.session_state:
|
||||
st.session_state.dp = DP(dst=st.session_state.dst)
|
||||
|
||||
|
@ -7,12 +7,18 @@ from streamlit_chat import message
|
||||
from .stt import show_voice_input
|
||||
from .tts import show_audio_player
|
||||
|
||||
from AMUseBotBackend.src.DP.dp import DP
|
||||
from AMUseBotBackend.src.DST.dst import DST
|
||||
from AMUseBotBackend.src.NLU.nlu import NLU
|
||||
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
INTENT_DICT_PATH = os.getenv('INTENT_DICT_PATH')
|
||||
MODEL_IDENTIFIER_PATH = os.getenv('MODEL_IDENTIFIER_PATH')
|
||||
|
||||
@st.cache_resource
|
||||
def get_nlu_model(intent_dict_path = 'AMUseBotFront/ai_talks/AMUseBotBackend/utils/intent_dict.json', model_identifier_path = 'AMUseBotFront/ai_talks/AMUseBotBackend/models/NLU/roberta-base-cookdial.txt'):
|
||||
def get_nlu_model(intent_dict_path = INTENT_DICT_PATH, model_identifier_path = MODEL_IDENTIFIER_PATH):
|
||||
return NLU(intent_dict_path=intent_dict_path,
|
||||
model_identifier_path=model_identifier_path)
|
||||
|
||||
|
@ -10,3 +10,12 @@ watchdog>=3.0.0
|
||||
setuptools~=65.5.0
|
||||
graphviz~=0.20.1
|
||||
Pillow~=9.5.0
|
||||
pandas==1.5.2
|
||||
scikit_learn==1.2.0
|
||||
simpletransformers==0.63.9
|
||||
torch==1.11.0
|
||||
spacy==3.5.0
|
||||
rank_bm25==0.2.2
|
||||
tqdm==4.64.1
|
||||
nlp==0.4.0
|
||||
python-dotenv==1.0.0
|
Loading…
Reference in New Issue
Block a user