diff --git a/requirements.txt b/requirements.txt index 565e922..ed10b89 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ flair==0.13.1 conllu==4.5.3 pandas==1.5.3 -numpy==1.26.4 -torch==2.3.0 \ No newline at end of file +numpy==1.24.4 +torch==2.3.0 +fuzzywuzzy==0.18.0 \ No newline at end of file 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