diff --git a/DM.py b/DM.py index 7f5a417..eb6adc5 100644 --- a/DM.py +++ b/DM.py @@ -49,5 +49,7 @@ class DM: return "findDataset" elif self.curr_state['intent'] == 'SUMMARIZEPAPER': return "sumPaper" + else: + return "ConvGen" else: return "Clarify"