From 71f525ec072451cdb36f0d4e2aab17540ddfc5a2 Mon Sep 17 00:00:00 2001 From: s495728 Date: Sun, 9 Jun 2024 14:52:22 +0200 Subject: [PATCH] Change requirements --- requirements.txt | 5 +++-- src/service/dialog_policy.py | 4 +++- 2 files changed, 6 insertions(+), 3 deletions(-) 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