loubnabnl HF staff commited on
Commit
baca49b
1 Parent(s): fc9416e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -123,7 +123,7 @@ with demo:
123
  run = gr.Button()
124
  output = gr.HTML(label="Generated code")
125
 
126
- event = run.click(code_generation, [code, max_new_tokens, temperature, seed], output)
127
  gr.HTML(label="Contact", value="<img src='https://huggingface.co/datasets/bigcode/admin/resolve/main/bigcode_contact.png' alt='contact' style='display: block; margin: auto; max-width: 800px;'>")
128
 
129
  demo.launch()
 
123
  run = gr.Button()
124
  output = gr.HTML(label="Generated code")
125
 
126
+ event = run.click(code_generation, [code, max_new_tokens, temperature, seed], output, api_name="predict")
127
  gr.HTML(label="Contact", value="<img src='https://huggingface.co/datasets/bigcode/admin/resolve/main/bigcode_contact.png' alt='contact' style='display: block; margin: auto; max-width: 800px;'>")
128
 
129
  demo.launch()