Update app.py

app.py

@@ -8,7 +8,6 @@ with gr.Blocks() as demo:
     with gr.Row():
         _, colum_2, _ = gr.Column(scale=1), gr.Column(scale=6), gr.Column(scale=1)
         with colum_2:
-            gr.Markdown(value=description)
             code = gr.Textbox(lines=5, label="Input code", value="def all_odd_elements(sequence):\n    \"\"\"Returns every odd element of the sequence.\"\"\"")
             
             with gr.Accordion("Advanced settings", open=False):
@@ -33,8 +32,7 @@ with gr.Blocks() as demo:
                     label="Random seed to use for the generation"
                 )
             run = gr.Button()
-            output = gr.HTML(label="Generated code")
 
-    event = run.click(code_generation, [code, max_new_tokens, temperature, seed], output, api_name="predict")
+    event = run.click(code_generation, [code, max_new_tokens, temperature, seed], code, api_name="predict")
 
 demo.launch()