Spaces:
Runtime error
Runtime error
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 β¨")
|