Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
 from transformers import pipeline
 import os
 
-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>"""
+description = """# <p style="text-align: center; color: white;"> 🎅 <span style='color: #ff75b3;'>SantaCoder:</span> Code Generation </p>
+<span style='color: white;'>This is a demo to generate code with <a href="https://huggingface.co/bigcode/santacoder" style="color: #ff75b3;">SantaCoder</a>, a 1.1B model for code generation in Python, Java & JavaScript.</span>"""
 
 token = os.environ["HUB_TOKEN"]
 device="cuda:0"
@@ -30,26 +30,29 @@ with demo:
     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"
-            )
+            
+            with gradio.Accordion("Advanced settings"):
+                max_tokens= gr.Slider(
+                    minimum=8,
+                    maximum=1024,
+                    step=1,
+                    value=16,
+                    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()
-        with gr.Column():
             output = gr.Textbox(lines=10, label="Generated code")
 
     event = run.click(code_generation, [code, max_tokens, temperature, seed], output)