Update app.py

app.py

@@ -22,6 +22,14 @@ model = AutoModelForCausalLM.from_pretrained(
 model.to(device)
 model.eval()
 
+# ==========================
+#  Example prompt from the paper
+# ==========================
+
+PAPER_PROMPT = (
+    "def hash_(listing_id):\n"
+    "    \"\"\"Creates an hashed column using the listing id for the vehicle\"\"\"\n"
+)
 
 # ==========================
 #  Prompt builder
@@ -105,10 +113,20 @@ with gr.Blocks(title="Python Code Generator") as demo:
         with gr.Column(scale=2):
             instruction_input = gr.Textbox(
                 label="Instruction",
-                placeholder="Describe the code you want. E.g., 'Write a Python function that checks if a number is prime.'",
+                placeholder=(
+                    "Describe the code you want. "
+                    "E.g., 'Write a Python function that checks if a number is prime.'"
+                ),
                 lines=4,
             )
 
+            # 🔹 Example prompt from the paper (click to fill the textbox)
+            gr.Examples(
+                examples=[[PAPER_PROMPT]],
+                inputs=[instruction_input],
+                label="Try the paper's hash_ example",
+            )
+
             is_first = gr.State(True)
 
             submit_btn = gr.Button("Generate Code")