update examples

app.py

@@ -9,13 +9,18 @@ This dataset includes Python code accompanied by docstrings that explain it. The
 Limitations: The model was fine-tuned on a small dataset from Jupyter Notebooks, so it can only explain simple,\
 common functions that are found in these notebooks, in a similar fashion to the text in markdown cells."
 
-EXAMPLE_1 = "def sort_function(arr):\n    n = len(arr)\n \n    # Traverse through all array elements\n    for i in range(n):\n \n        # Last i elements are already in place\n        for j in range(0, n-i-1):\n \n            # traverse the array from 0 to n-i-1\n            # Swap if the element found is greater\n            # than the next element\n            if arr[j] > arr[j+1]:\n                arr[j], arr[j+1] = arr[j+1], arr[j]"
-EXAMPLE_2 =  "from sklearn import model_selection\nX_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=0.2)"
-EXAMPLE_3 = "def load_text(file)\n    with open(filename, 'r') as f:\n        text = f.read()\n    return text"
+EXAMPLE_1 =  "from sklearn import model_selection\nX_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=0.2)"
+EXAMPLE_2 = "def load_text(file)\n    with open(filename, 'r') as f:\n        text = f.read()\n    return text"
+EXAMPLE_3 = "net.zero_grad()\nloss.backward()"
+EXAMPLE_4 = "net.zero_grad()\nloss.backward()\n\nnoptimizer.step()"
+EXAMPLE_5 = "def sort_function(arr):\n    n = len(arr)\n \n    # Traverse through all array elements\n    for i in range(n):\n \n        # Last i elements are already in place\n        for j in range(0, n-i-1):\n \n            # traverse the array from 0 to n-i-1\n            # Swap if the element found is greater\n            # than the next element\n            if arr[j] > arr[j+1]:\n                arr[j], arr[j+1] = arr[j+1], arr[j]"
+
 example = [
-    [EXAMPLE_1, 32, 0.6, 42],
-    [EXAMPLE_2, 16, 0.6, 42],
-    [EXAMPLE_3, 11, 0.2, 42],
+    [EXAMPLE_1, 34, 0.6, 42],
+    [EXAMPLE_2, 11, 0.6, 42],
+    [EXAMPLE_3, 30, 0.6, 42],
+    [EXAMPLE_4, 46, 0.6, 42],
+    [EXAMPLE_5, 32, 0.6, 42],
     ]
 
 device="cuda:0"