💄 improve examples and params

Signed-off-by: peter szemraj <[email protected]>

app.py

@@ -1,5 +1,3 @@
-import random
-
 import gradio as gr
 import torch
 from gradio.themes.utils import sizes
@@ -57,75 +55,78 @@ def run_inference(
     outputs = model.generate(
         **inputs,
         do_sample=True,
-        early_stopping=True,
+        # early_stopping=True,
         max_new_tokens=max_new_tokens,
-        min_new_tokens=8,
+        min_new_tokens=2,
         no_repeat_ngram_size=6,
-        num_beams=3,
         renormalize_logits=True,
         repetition_penalty=repetition_penalty,
         temperature=temperature,
         top_p=top_p,
     )
-    text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+    text = tokenizer.batch_decode(
+        outputs,
+        skip_special_tokens=True,
+    )[0]
     return text
 
 
 examples = [
-    ["def add_numbers(a, b):\n    return", 0.2, 96, 0.9, 1.2],
     [
-        "class Car:\n    def __init__(self, make, model):\n        self.make = make\n        self.model = model\n\n    def display_car(self):",
+        'def greet(name: str) -> None:\n    """\n    Greets the user\n    """\n    print(f"Hello,',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
     [
-        "import pandas as pd\ndata = {'Name': ['Tom', 'Nick', 'John'], 'Age': [20, 21, 19]}\ndf = pd.DataFrame(data).convert_dtypes()\n# eda",
+        'for i in range(5):\n    """\n    Loop through 0 to 4\n    """\n    print(i,',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
+    ['x = 10\n"""Check if x is greater than 5"""\nif x > 5:', 0.2, 64, 0.9, 1.2],
+    ["def square(x: int) -> int:\n    return", 0.2, 64, 0.9, 1.2],
+    ['import math\n"""Math operations"""\nmath.', 0.2, 64, 0.9, 1.2],
     [
-        "def factorial(n):\n    if n == 0:\n        return 1\n    else:",
+        'def is_even(n) -> bool:\n    """\n    Check if a number is even\n    """\n    if n % 2 == 0:',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
     [
-        'def fibonacci(n):\n    if n <= 0:\n        raise ValueError("Incorrect input")\n    elif n == 1:\n        return 0\n    elif n == 2:\n        return 1\n    else:',
+        'while True:\n    """Infinite loop example"""\n    print("Infinite loop,',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
     [
-        "import matplotlib.pyplot as plt\nimport numpy as np\nx = np.linspace(0, 10, 100)\n# simple plot",
+        "def sum_list(lst: list[int]) -> int:\n    total = 0\n    for item in lst:",
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
-    ["def reverse_string(s:str) -> str:\n    return", 0.2, 96, 0.9, 1.2],
-    ["def is_palindrome(word:str) -> bool:\n    return", 0.2, 96, 0.9, 1.2],
     [
-        "def bubble_sort(lst: list):\n    n = len(lst)\n    for i in range(n):\n        for j in range(0, n-i-1):",
+        'try:\n    """\n    Exception handling\n    """\n    x = int(input("Enter a number: "))\nexcept ValueError:',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
     [
-        "def binary_search(arr, low, high, x):\n    if high >= low:\n        mid = (high + low) // 2\n        if arr[mid] == x:\n            return mid\n        elif arr[mid] > x:",
+        'def divide(a: float, b: float) -> float:\n    """\n    Divide a by b\n    """\n    if b != 0:',
         0.2,
-        96,
+        64,
         0.9,
         1.2,
     ],
 ]
 
+
 # Define the Gradio Blocks interface
 with gr.Blocks(theme=theme, analytics_enabled=False, css=_styles) as demo:
     with gr.Column():
@@ -133,7 +134,7 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=_styles) as demo:
         with gr.Row():
             with gr.Column():
                 instruction = gr.Textbox(
-                    value=random.choice([e[0] for e in examples]),
+                    value=examples[0][0],
                     placeholder="Enter your code here",
                     label="Code",
                     elem_id="q-input",
@@ -176,7 +177,7 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=_styles) as demo:
                                     )
                                     repetition_penalty = gr.Slider(
                                         label="Repetition penalty",
-                                        value=1.1,
+                                        value=1.2,
                                         minimum=1.0,
                                         maximum=2.0,
                                         step=0.05,