Stanislas commited on
Commit
3a46326
β€’
1 Parent(s): 7d2ff94

Update app

Browse files
Files changed (1) hide show
  1. app.py +18 -8
app.py CHANGED
@@ -56,13 +56,22 @@ def main():
56
  with gr.Blocks() as demo:
57
  gr.Markdown(
58
  """# CodeGeeX: A Multilingual Code Generation Model
59
- <img src="https://github.com/THUDM/CodeGeeX/blob/main/resources/logo/codegeex_logo.png">
 
 
 
 
 
 
 
 
 
60
  We introduce CodeGeeX, a large-scale multilingual code generation model with 13 billion parameters, pre-trained on a large code corpus of more than 20 programming languages. CodeGeeX was trained on more than 850 billion tokens on a cluster of 1,536 [Ascend 910 AI Processors](https://e.huawei.com/en/products/servers/ascend). CodeGeeX supports 15+ programming languages for both code generation and code translation. CodeGeeX is open source, please refer to our [GitHub](https://github.com/THUDM/CodeGeeX) for more details. This is a minimal-functional DEMO, for other DEMOs like code translation, please visit our [Homepage](https://models.aminer.cn/codegeex/). We also offer a free [VS Code extension](https://marketplace.visualstudio.com/items?itemName=aminer.codegeex) for full functionality.
61
  """)
62
 
63
  with gr.Row():
64
  with gr.Column():
65
- prompt = gr.Textbox(lines=13, placeholder='Input', label='Input')
66
  with gr.Row():
67
  gen = gr.Button("Generate")
68
  clr = gr.Button("Clear")
@@ -81,11 +90,13 @@ def main():
81
  "SQL", "Kotlin", "R", "Fortran"], value='lang', label='Programming Language',
82
  default="Python")
83
  with gr.Column():
84
- seed = gr.Slider(maximum=10000, value=43, step=1, label='Seed')
85
- out_seq_length = gr.Slider(maximum=1024, value=256, minimum=1, step=1, label='Output Sequence Length')
86
- temperature = gr.Slider(maximum=1, value=0.9, minimum=0, label='Temperature')
87
- top_k = gr.Slider(maximum=40, value=0, minimum=0, step=1, label='Top K')
88
- top_p = gr.Slider(maximum=1, value=1.0, minimum=0, label='Top P')
 
 
89
 
90
  inputs = [prompt, lang, seed, out_seq_length, temperature, top_k, top_p]
91
  gen.click(fn=predict, inputs=inputs, outputs=outputs)
@@ -94,7 +105,6 @@ def main():
94
  gr_examples = gr.Examples(examples=examples, inputs=[prompt, lang],
95
  label="Example Inputs (Click to insert an examplet it into the input box)",
96
  examples_per_page=20)
97
- gr.Markdown("![visitors](https://visitor-badge.glitch.me/badge?page_id=THUDM.CodeGeeX)")
98
 
99
  demo.launch()
100
 
 
56
  with gr.Blocks() as demo:
57
  gr.Markdown(
58
  """# CodeGeeX: A Multilingual Code Generation Model
59
+ <img src="https://raw.githubusercontent.com/THUDM/CodeGeeX/main/resources/logo/codegeex_logo.png">
60
+ """)
61
+ gr.Markdown(
62
+ """
63
+ <p align="center">
64
+ 🏠 <a href="https://models.aminer.cn/codegeex" target="_blank">Homepage</a> | πŸ“– <a href="http://keg.cs.tsinghua.edu.cn/codegeex/" target="_blank">Blog</a> | πŸͺ§ <a href="https://models.aminer.cn/codegeex/playground" target="_blank">DEMO</a> | πŸ›  <a href="https://marketplace.visualstudio.com/items?itemName=aminer.codegeex" target="_blank">VS Code Extension</a> | πŸ’» <a href="https://github.com/THUDM/CodeGeeX" target="_blank">Source code</a> | πŸ€– <a href="https://models.aminer.cn/codegeex/download/request" target="_blank">Download Model</a>
65
+ </p>
66
+ """)
67
+ gr.Markdown(
68
+ """
69
  We introduce CodeGeeX, a large-scale multilingual code generation model with 13 billion parameters, pre-trained on a large code corpus of more than 20 programming languages. CodeGeeX was trained on more than 850 billion tokens on a cluster of 1,536 [Ascend 910 AI Processors](https://e.huawei.com/en/products/servers/ascend). CodeGeeX supports 15+ programming languages for both code generation and code translation. CodeGeeX is open source, please refer to our [GitHub](https://github.com/THUDM/CodeGeeX) for more details. This is a minimal-functional DEMO, for other DEMOs like code translation, please visit our [Homepage](https://models.aminer.cn/codegeex/). We also offer a free [VS Code extension](https://marketplace.visualstudio.com/items?itemName=aminer.codegeex) for full functionality.
70
  """)
71
 
72
  with gr.Row():
73
  with gr.Column():
74
+ prompt = gr.Textbox(lines=13, placeholder='Please enter the description or select an example input below.',label='Input')
75
  with gr.Row():
76
  gen = gr.Button("Generate")
77
  clr = gr.Button("Clear")
 
90
  "SQL", "Kotlin", "R", "Fortran"], value='lang', label='Programming Language',
91
  default="Python")
92
  with gr.Column():
93
+ seed = gr.Slider(maximum=10000, value=8888, step=1, label='Seed')
94
+ with gr.Row():
95
+ out_seq_length = gr.Slider(maximum=1024, value=128, minimum=1, step=1, label='Output Sequence Length')
96
+ temperature = gr.Slider(maximum=1, value=0.2, minimum=0, label='Temperature')
97
+ with gr.Row():
98
+ top_k = gr.Slider(maximum=40, value=0, minimum=0, step=1, label='Top K')
99
+ top_p = gr.Slider(maximum=1, value=1.0, minimum=0, label='Top P')
100
 
101
  inputs = [prompt, lang, seed, out_seq_length, temperature, top_k, top_p]
102
  gen.click(fn=predict, inputs=inputs, outputs=outputs)
 
105
  gr_examples = gr.Examples(examples=examples, inputs=[prompt, lang],
106
  label="Example Inputs (Click to insert an examplet it into the input box)",
107
  examples_per_page=20)
 
108
 
109
  demo.launch()
110