diff --git a/requirements.txt b/requirements.txt index 75af938..232fa7f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ flair==0.13.1 conllu==4.5.3 pandas==1.5.3 -numpy==1.26.4 torch==2.3.0 -convlab==3.0.2a0 \ No newline at end of file +convlab==3.0.2a0 +numpy==1.24.4 +fuzzywuzzy==0.18.0 diff --git a/src/service/dialog_policy.py b/src/service/dialog_policy.py index cafa615..12eaacc 100644 --- a/src/service/dialog_policy.py +++ b/src/service/dialog_policy.py @@ -56,4 +56,6 @@ class SimpleRulePolicy(Policy): if domain in ["pizza", "drink"]: system_action[(domain, 'Recommend')].append(['Name', choice['name']]) if domain in ["size"]: - system_action[(domain, 'Recommend')].append(['Size', choice['size']]) \ No newline at end of file + system_action[(domain, 'Recommend')].append(['Size', choice['size']]) + +dialogPolicy = SimpleRulePolicy() \ No newline at end of file