Update app.py

app.py

@@ -62,7 +62,7 @@ def execute_code(code, test):
     output = new_stdout.getvalue()
     return output
 
-example = """
+examples = ["""
 def prime_factors(n):
     i = 2
     factors = []
@@ -75,12 +75,24 @@ def prime_factors(n):
     if n > 1:
         factors.append(n)
     return factors
-    """
+    """,
+"""
+import numpy
+    def matrix_multiplication(A, B):
+    return np.dot(A, B)
+"""
+    ]
+example = examples[0]
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Llama_test: generate unit test for your Python code</center></h1>")
     with gr.Row():
         code_input = gr.Code(example, label="Provide the code of the function you want to test")
+        gr.Examples(
+        examples=examples,
+        inputs=code_input,
+    )
+
     generate_btn = gr.Button("Generate test")
     with gr.Row():
         code_output = gr.Code()