stillerman commited on
Commit
8284e69
1 Parent(s): 39a889e

changed demo text and prompt

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -4,8 +4,8 @@ from transformers import pipeline
4
  import os
5
  import torch
6
 
7
- description = """# <p style="text-align: center; color: white;"> 🎅 <span style='color: #ff75b3;'>SantaCoder:</span> Code Generation </p>
8
- <span style='color: white;'>This is a demo to generate code with <a href="https://huggingface.co/bigcode/santacoder" style="color: #ff75b3;">SantaCoder</a>,
9
  a 1.1B parameter model for code generation in Python, Java & JavaScript. The model can also do infilling, just specify where you would like the model to complete code
10
  with the <span style='color: #ff75b3;'>&lt;FILL-HERE&gt;</span> token.</span>"""
11
 
@@ -101,8 +101,7 @@ with demo:
101
  if n <= 1
102
  n
103
  else
104
- <FILL-HERE>
105
- end''')
106
 
107
  with gr.Accordion("Advanced settings", open=False):
108
  max_new_tokens= gr.Slider(
@@ -123,7 +122,6 @@ end''')
123
  minimum=0,
124
  maximum=1000,
125
  step=1,
126
- value=1,
127
  label="Random seed to use for the generation"
128
  )
129
  run = gr.Button()
 
4
  import os
5
  import torch
6
 
7
+ description = """# <p style="text-align: center; color: white;"> 🎅 <span style='color: #ff75b3;'>SantaCoder-Ruby:</span> Code Generation </p>
8
+ <span style='color: white;'>This is a demo to generate code with <a href="https://huggingface.co/stillerman/santacoder-ruby" style="color: #ff75b3;">SantaCoder-Ruby</a> which is a fine-tuned version of <a href="https://huggingface.co/bigcode/santacoder" style="color: #ff75b3;">SantaCoder</a>,
9
  a 1.1B parameter model for code generation in Python, Java & JavaScript. The model can also do infilling, just specify where you would like the model to complete code
10
  with the <span style='color: #ff75b3;'>&lt;FILL-HERE&gt;</span> token.</span>"""
11
 
 
101
  if n <= 1
102
  n
103
  else
104
+ <FILL-HERE>''')
 
105
 
106
  with gr.Accordion("Advanced settings", open=False):
107
  max_new_tokens= gr.Slider(
 
122
  minimum=0,
123
  maximum=1000,
124
  step=1,
 
125
  label="Random seed to use for the generation"
126
  )
127
  run = gr.Button()