update

pages/summarization_playground.py

@@ -61,7 +61,7 @@ def process_input(input_text, model_selection, prompt):
 def update_input(example):
     if example == random_label:
         datapoint = random.choice(dataset)
-        return datapoint['section_text'] + '\n\nDialogue:' + datapoint['dialogue']
+        return datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue']
     return examples[example]
 
 def create_summarization_interface():
@@ -73,8 +73,8 @@ def create_summarization_interface():
             model_dropdown = gr.Dropdown(choices=Model.__model_list__, label="Choose a model", value=Model.__model_list__[0])
         
         Template_text = gr.Textbox(value="""Summarize the following dialogue""", label='Input Prompting Template', lines=8, placeholder='Input your prompts')
-
-        input_text = gr.Textbox(label="Input Dialogue", lines=10, placeholder="Enter text here...", value=random.choice(dataset)['dialogue'])
+        datapoint = random.choice(dataset)
+        input_text = gr.Textbox(label="Input Dialogue", lines=10, placeholder="Enter text here...", value=datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue'])
         submit_button = gr.Button("✨ Submit ✨")
         output = gr.Markdown()
 

utils/model.py

@@ -55,7 +55,6 @@ class Model(torch.nn.Module):
             temperature=temp,
             num_return_sequences=1,
             eos_token_id=self.tokenizer.eos_token_id,
-            return_full_text=False
         )
         
         return sequences[-1]['generated_text']