lvwerra HF staff commited on
Commit
7389f5f
·
1 Parent(s): 4b21286

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -256,10 +256,10 @@ description = """
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>
262
- <li><a href="https://huggingface.co/lvwerra/starcoderbase-gsm8k">StarCoderBase GSM8K</a>: A finetuned version of StarCoderBase on on the GSM8K dataset using reinforcement learning via TRL's <a href="https://github.com/huggingface/trl/pull/424">TRL's TextEnvironment</a>.</li>
263
  </ul>
264
  </div>
265
  """
@@ -290,7 +290,13 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
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):
@@ -347,11 +353,6 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
347
  outputs=[system_prompt],
348
  )
349
 
350
- gr_examples = gr.Examples(
351
- examples=[example for client in clients.values() for example in client[3]],
352
- inputs=[instruction],
353
- cache_examples=False,
354
- )
355
 
356
 
357
  submit.click(
 
256
  </div>
257
  <div style="text-align: left;">
258
  <hr>
259
+ <p>This is a demo to generate text the following StarCoderBase models fine-tuned using <a href="https://github.com/huggingface/trl/pull/424">TRL's TextEnvironment</a>:</p>
260
  <ul>
261
+ <li><a href="https://huggingface.co/vwxyzjn/starcoderbase-triviaqa">StarCoderBase TriviaQA</a>: Uses a Wikipedia search index to answer trivia questions. It was trained on the TriviaQA dataset.</li>
262
+ <li><a href="https://huggingface.co/lvwerra/starcoderbase-gsm8k">StarCoderBase GSM8K</a>: Uses a Python Interpreter to answer math questions. It was trained on the GSM8K dataset.</li>
263
  </ul>
264
  </div>
265
  """
 
290
  label="Output",
291
  color_map={"query": "red", "call": "green", "response": "blue", "submit": "yellow", "model": "pink"},
292
  )
293
+
294
+ gr_examples = gr.Examples(
295
+ examples=[example for client in clients.values() for example in client[3]],
296
+ inputs=[instruction],
297
+ cache_examples=False,
298
+ )
299
+
300
  with gr.Row():
301
  with gr.Column():
302
  with gr.Accordion("Raw output", open=False):
 
353
  outputs=[system_prompt],
354
  )
355
 
 
 
 
 
 
356
 
357
 
358
  submit.click(