Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -255,6 +255,7 @@ description = """
|
|
255 |
<img src="https://huggingface.co/datasets/trl-internal-testing/example-images/resolve/main/images/textenv_demo_banner.png">
|
256 |
</div>
|
257 |
<div style="text-align: left;">
|
|
|
258 |
<p>This is a demo to generate text and code with the following StarCoderBase models:</p>
|
259 |
<ul>
|
260 |
<li><a href="https://huggingface.co/vwxyzjn/starcoderbase-triviaqa">StarCoderBase TriviaQA</a>: A finetuned version of StarCoderBase on on the TriviaQA dataset using reinforcement learning via <a href="https://github.com/huggingface/trl/pull/424">TRL's TextEnvironment</a>.</li>
|
@@ -284,14 +285,16 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
|
|
284 |
elem_id="q-input",
|
285 |
)
|
286 |
submit = gr.Button("Generate", variant="primary")
|
287 |
-
|
288 |
output = gr.HighlightedText(
|
289 |
label="Output",
|
290 |
color_map={"query": "red", "call": "green", "response": "blue", "submit": "yellow", "model": "pink"},
|
291 |
)
|
292 |
-
|
293 |
with gr.Row():
|
294 |
with gr.Column():
|
|
|
|
|
295 |
with gr.Accordion("Advanced settings", open=False):
|
296 |
with gr.Row():
|
297 |
column_1, column_2 = gr.Column(), gr.Column()
|
|
|
255 |
<img src="https://huggingface.co/datasets/trl-internal-testing/example-images/resolve/main/images/textenv_demo_banner.png">
|
256 |
</div>
|
257 |
<div style="text-align: left;">
|
258 |
+
<hr>
|
259 |
<p>This is a demo to generate text and code with the following StarCoderBase models:</p>
|
260 |
<ul>
|
261 |
<li><a href="https://huggingface.co/vwxyzjn/starcoderbase-triviaqa">StarCoderBase TriviaQA</a>: A finetuned version of StarCoderBase on on the TriviaQA dataset using reinforcement learning via <a href="https://github.com/huggingface/trl/pull/424">TRL's TextEnvironment</a>.</li>
|
|
|
285 |
elem_id="q-input",
|
286 |
)
|
287 |
submit = gr.Button("Generate", variant="primary")
|
288 |
+
|
289 |
output = gr.HighlightedText(
|
290 |
label="Output",
|
291 |
color_map={"query": "red", "call": "green", "response": "blue", "submit": "yellow", "model": "pink"},
|
292 |
)
|
293 |
+
|
294 |
with gr.Row():
|
295 |
with gr.Column():
|
296 |
+
with gr.Accordion("Raw output", open=False):
|
297 |
+
output2 = gr.Code(elem_id="q-output", lines=30, label="Raw output")
|
298 |
with gr.Accordion("Advanced settings", open=False):
|
299 |
with gr.Row():
|
300 |
column_1, column_2 = gr.Column(), gr.Column()
|