Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
Β·
47c8870
1
Parent(s):
af1a6de
fix
Browse files
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
|
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 β¨")
|