Stanislas commited on
Commit
aa544c1
1 Parent(s): 3a46326
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -18,15 +18,15 @@ def predict(prompt, lang, seed, out_seq_length, temperature, top_k, top_p):
18
  url = 'https://tianqi.aminer.cn/api/v2/multilingual_code_generate_block'
19
 
20
  payload = json.dumps({
21
- "apikey" : APIKEY,
22
- "apisecret" : APISECRET,
23
- "prompt" : prompt,
24
- "lang" : lang,
25
- "out_seq_lengt": out_seq_length,
26
- "seed" : seed,
27
- "temperature" : temperature,
28
- "top_k" : top_k,
29
- "top_p" : top_p
30
  })
31
 
32
  headers = {
@@ -55,7 +55,7 @@ def main():
55
 
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(
@@ -66,7 +66,7 @@ def main():
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():
 
18
  url = 'https://tianqi.aminer.cn/api/v2/multilingual_code_generate_block'
19
 
20
  payload = json.dumps({
21
+ "apikey" : APIKEY,
22
+ "apisecret" : APISECRET,
23
+ "prompt" : prompt,
24
+ "lang" : lang,
25
+ "out_seq_length": out_seq_length,
26
+ "seed" : seed,
27
+ "temperature" : temperature,
28
+ "top_k" : top_k,
29
+ "top_p" : top_p,
30
  })
31
 
32
  headers = {
 
55
 
56
  with gr.Blocks() as demo:
57
  gr.Markdown(
58
+ """
59
  <img src="https://raw.githubusercontent.com/THUDM/CodeGeeX/main/resources/logo/codegeex_logo.png">
60
  """)
61
  gr.Markdown(
 
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 supports 15+ programming languages for both code generation and 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():