Change requirements

2024-06-09 14:52:22 +02:00 · 2024-06-09 14:52:22 +02:00 · 71f525ec07
commit 71f525ec07
parent 312225b307
2 changed files with 6 additions and 3 deletions
--- 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
+numpy==1.24.4
+torch==2.3.0
+fuzzywuzzy==0.18.0
--- 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']])
+                    system_action[(domain, 'Recommend')].append(['Size', choice['size']])
+
+dialogPolicy = SimpleRulePolicy()