update dataset

pages/summarization_playground.py

@@ -34,7 +34,7 @@ Back in Boston, Kidd is going to rely on Lively even more. He'll play close to 3
 }
 
 def generate_answer(sources, model_name, prompt):
-    content = prompt + '\n{' + sources + '}\n\n'
+    content = prompt + '\n{' + sources + '}\n\nsummary:'
     global __model_on_gpu__
 
     if __model_on_gpu__ != model_name:
@@ -60,7 +60,8 @@ def process_input(input_text, model_selection, prompt):
 
 def update_input(example):
     if example == random_label:
-        return random.choice(dataset)['dialogue']
+        datapoint = random.choice(dataset)
+        return datapoint['section_text'] + '\n\nDialogue:' + datapoint['dialogue']
     return examples[example]
 
 def create_summarization_interface():

utils/data.py

@@ -1,4 +1,4 @@
 from datasets import load_dataset 
-dialogsum = load_dataset('knkarthick/dialogsum')
-dataset = list(dialogsum['train']) + list(dialogsum['validation']) + list(dialogsum['test'])
+dialogsum = load_dataset('har1/MTS_Dialogue-Clinical_Note')
+dataset = list(dialogsum['train'])