Update app.py

app.py

@@ -35,14 +35,14 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=devi
 def post_processing(prompt, completion):
     completion = "<span style='color: #ff75b3;'>" + completion + "</span>"
     prompt = "<span style='color: #727cd6;'>" + prompt + "</span>"
-    code_html = f"<br><hr><br><pre style='max-width: 500px; margin: 0 auto; display: block;'><code>{prompt}{completion}</code></pre><br><hr>"
+    code_html = f"<br><hr><br><pre style='font-size: 12px'><code>{prompt}{completion}</code></pre><br><hr>"
     return GENERATION_TITLE + code_html
 
 def post_processing_fim(prefix, middle, suffix):
     prefix = "<span style='color: #727cd6;'>" + prefix + "</span>"
     middle = "<span style='color: #ff75b3;'>" + middle + "</span>"
     suffix = "<span style='color: #727cd6;'>" + suffix + "</span>"
-    code_html = f"<br><hr><br><pre style='max-width: 500px; margin: 0 auto; display: block;'><code>{prefix}{middle}{suffix}</code></pre><br><hr>"
+    code_html = f"<br><hr><br><pre style='font-size: 12px'><code>{prefix}{middle}{suffix}</code></pre><br><hr>"
     return GENERATION_TITLE + code_html
 
 def fim_generation(prompt, max_new_tokens, temperature):
@@ -97,14 +97,14 @@ with demo:
         _, 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")
+            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=32,
+                    value=48,
                     label="Number of tokens to generate",
                 )
                 temperature = gr.Slider(