Update app.py

app.py

@@ -1,3 +1,40 @@
 import gradio as gr
 
-gr.Interface(lambda x:x, "code", "code").launch()
+def code_generation(prompt, max_new_tokens, temperature=0.2, seed=42):
+    #set_seed(seed)
+    return prompt + "\n\n" + prompt
+
+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):
+                max_new_tokens= gr.Slider(
+                    minimum=8,
+                    maximum=1024,
+                    step=1,
+                    value=48,
+                    label="Number of tokens to generate",
+                )
+                temperature = gr.Slider(
+                    minimum=0.1,
+                    maximum=2.5,
+                    step=0.1,
+                    value=0.2,
+                    label="Temperature",
+                )
+                seed = gr.Slider(
+                    minimum=0,
+                    maximum=1000,
+                    step=1,
+                    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")
+
+demo.launch()