Update app.py

app.py

@@ -23,7 +23,7 @@ model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 model.to(device)
 tokenizer = Tokenizer.from_pretrained(tokenizer_path)
 
-def greet(prompt):
+def generate(prompt):
     model_input = torch.tensor(tokenizer.encode(prompt), dtype=torch.long, device=device).view((1, len(prompt)))
     return tokenizer.decode(model.generate(model_input, max_new_tokens=4, context_size=context_size)[0].tolist())
 
@@ -41,6 +41,15 @@ with gr.Blocks() as demo:
     output = gr.Text(label="Next turn", interactive=False)
 
     submit = gr.Button("Submit")
-    submit.click(greet, [prompt], [output])
-
+    submit.click(generate, [prompt], [output])
+
+    gr.Examples(
+        [
+            ["1. e4", ],
+            ["1. e4 g6 2."],
+        ],
+        inputs=[prompt],
+        outputs=[outpu],
+        fn=generate
+    )
 demo.launch()