Update app.py

app.py

@@ -30,7 +30,7 @@ def get_completion(code_snippet):
     modify_list(my_list)
     print(my_list)
     Correct output: [0, 4]
-    Explanation: input_list is modified by appending 4, but the reassignment does not affect the original list.
+    Explanation: The function appends 4 to input_list, which modifies my_list directly. The reassignment does not affect the original list.
 
     ---------------------
     Example 3:
@@ -38,6 +38,31 @@ def get_completion(code_snippet):
     Correct output: hello
     Explanation: This line outputs the string 'hello' to the console.
     ---------------------
+    Example 4:
+    for i in range(5):
+        print(i)
+    Correct output: 
+    0
+    1
+    2
+    3
+    4
+    Explanation: This loop iterates from 0 to 4, printing each number on a new line.
+    ---------------------
+    Example 5:
+    def add(a, b):
+        return a + b
+    result = add(3, 4)
+    print(result)
+    Correct output: 7
+    Explanation: The function add takes two arguments, adds them together, and returns the result. print(result) outputs 7.
+    ---------------------
+    Example 6:
+    if True:
+        print("This is true")
+    Correct output: This is true
+    Explanation: The condition True always evaluates to True, so the print statement is executed.
+    ---------------------
     """
     
     prompt = f"""
@@ -52,7 +77,7 @@ def get_completion(code_snippet):
 
     try:
         print("Generating text...")
-        response = text_generator(prompt, max_length=300, num_return_sequences=1)
+        response = text_generator(prompt, max_new_tokens=100, num_return_sequences=1)
         print("Text generation complete.")
         return response[0]['generated_text']
     except Exception as e: