s495726/chat-restaruacja
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change requirements

Browse Source
This commit is contained in:
s495728 2024-06-09 14:52:22 +02:00
parent 1cd62db966
commit 09157ce36f
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements.txt
View File

@ -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
convlab==3.0.2a0
numpy==1.24.4
fuzzywuzzy==0.18.0

4
src/service/dialog_policy.py
View File

@ -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()