kz209 commited on
Commit
b596b41
1 Parent(s): c9c9f16
Files changed (1) hide show
  1. pages/summarization_playground.py +2 -2
pages/summarization_playground.py CHANGED
@@ -32,7 +32,7 @@ Back in Boston, Kidd is going to rely on Lively even more. He'll play close to 3
32
  }
33
 
34
  def generate_answer(sources, model_name, prompt):
35
- content = prompt + '\n' + sources + '\n\n'
36
  global __model_on_gpu__
37
 
38
  if __model_on_gpu__ != model_name:
@@ -70,7 +70,7 @@ def create_summarization_interface():
70
 
71
  Template_text = gr.Textbox(value="""Summarize the following dialogue""", label='Input Prompting Template', lines=8, placeholder='Input your prompts')
72
 
73
- input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...", value=random.choice(dataset)['dialogue'])
74
  submit_button = gr.Button("✨ Submit ✨")
75
  output = gr.Markdown()
76
 
 
32
  }
33
 
34
  def generate_answer(sources, model_name, prompt):
35
+ content = prompt + '\n{' + sources + '}\n\n'
36
  global __model_on_gpu__
37
 
38
  if __model_on_gpu__ != model_name:
 
70
 
71
  Template_text = gr.Textbox(value="""Summarize the following dialogue""", label='Input Prompting Template', lines=8, placeholder='Input your prompts')
72
 
73
+ input_text = gr.Textbox(label="Input Dialogue", lines=10, placeholder="Enter text here...", value=random.choice(dataset)['dialogue'])
74
  submit_button = gr.Button("✨ Submit ✨")
75
  output = gr.Markdown()
76