kz209 commited on
Commit
47c8870
Β·
1 Parent(s): af1a6de
Files changed (1) hide show
  1. pages/summarization_playground.py +2 -2
pages/summarization_playground.py CHANGED
@@ -10,7 +10,7 @@ import torch
10
 
11
  load_dotenv()
12
 
13
- __model_on_gpu__ = ''
14
  model = {model_name: Model(model_name) for model_name in Model.__model_list__}
15
 
16
  random_label = 'πŸ”€ Random dialogue from dataset'
@@ -67,7 +67,7 @@ def create_summarization_interface():
67
  example_dropdown = gr.Dropdown(choices=list(examples.keys()), label="Choose an example")
68
  model_dropdown = gr.Dropdown(choices=Model.__model_list__, label="Choose a model", value=Model.__model_list__[0])
69
 
70
- Template_text = gr.Textbox(value="""Summariza the following dialogue""", label='Input Prompting Template', lines=8, placeholder='Input your prompts, must include \{sources\}')
71
 
72
  input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...")
73
  submit_button = gr.Button("✨ Submit ✨")
 
10
 
11
  load_dotenv()
12
 
13
+ __model_on_gpu__ = Model.__model_list__[0]
14
  model = {model_name: Model(model_name) for model_name in Model.__model_list__}
15
 
16
  random_label = 'πŸ”€ Random dialogue from dataset'
 
67
  example_dropdown = gr.Dropdown(choices=list(examples.keys()), label="Choose an example")
68
  model_dropdown = gr.Dropdown(choices=Model.__model_list__, label="Choose a model", value=Model.__model_list__[0])
69
 
70
+ Template_text = gr.Textbox(value="""Summariza the following dialogue""", label='Input Prompting Template', lines=8, placeholder='Input your prompts')
71
 
72
  input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...")
73
  submit_button = gr.Button("✨ Submit ✨")