Update space format

app.py

@@ -3,13 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
 from transformers import pipeline
 import os
 
-title = "Santa Model Generator"
-description = "Demo"
-example = [
-    ["def print_hello_world():", 8, 0.6, 42],
-    ["def get_file_size(filepath):", 24, 0.6, 42],
-    ["def count_lines(filename):", 40, 0.6, 42],
-    ["def count_words(filename):", 40, 0.6, 42]]
+description = """# <p style="text-align: center; color: white;"> ❄️ <span style='color: #ff75b3;'>IceCoder</span> Generator ❄️ </p>
+<span style='color: white;'>This is a demo to generate code with <a href="todo" style="color: #ff75b3;">IceCoder</a>, a 1.1B model for code generation in Python, Java & JavaScript.</span>"""
 
 token = os.environ["HUB_TOKEN"]
 device="cuda:0"
@@ -26,37 +21,38 @@ def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
     return generated_text
 
 
-iface = gr.Interface(
-    fn=code_generation, 
-    inputs=[
-        gr.Textbox(lines=10, label="Input code"),
-        gr.inputs.Slider(
-            minimum=8,
-            maximum=1000,
-            step=1,
-            default=8,
-            label="Number of tokens to generate",
-        ),
-        gr.inputs.Slider(
-            minimum=0,
-            maximum=2.5,
-            step=0.1,
-            default=0.6,
-            label="Temperature",
-        ),
-        gr.inputs.Slider(
-            minimum=0,
-            maximum=1000,
-            step=1,
-            default=42,
-            label="Random seed to use for the generation"
-        )
-    ],
-    outputs=gr.Textbox(label="Predicted code", lines=10),
-    examples=example,
-    layout="horizontal",
-    theme="peach",
-    description=description,
-    title=title
+demo = gr.Blocks(
+    css=".gradio-container {background-color: #20233fff; color:white}"
 )
-iface.launch()
+with demo:
+    with gr.Row():
+        gr.Markdown(value=description)
+    with gr.Row():
+        with gr.Column():
+            code = gr.Textbox(lines=10, label="Input code")
+            max_tokens= gr.Slider(
+                minimum=8,
+                maximum=1000,
+                step=1,
+                label="Number of tokens to generate",
+            )
+            temperature = gr.Slider(
+                minimum=0.1,
+                maximum=2.5,
+                step=0.1,
+                label="Temperature",
+            )
+            seed = gr.Slider(
+                minimum=0,
+                maximum=1000,
+                step=1,
+                label="Random seed to use for the generation"
+            )
+            run = gr.Button()
+        with gr.Column():
+            output = gr.Textbox(lines=10, label="Generated code")
+
+    event = run.click(code_generation, [code, max_tokens, temperature, seed], output)
+    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;'>")
+
+demo.launch()