Update app.py

app.py

@@ -114,6 +114,7 @@ with gr.Blocks(css='style.css') as demo:
         undo_button = gr.Button('↩️ Undo', variant='secondary')
         clear_button = gr.Button('🗑️  Clear', variant='secondary')
 
+    api_button = gr.Button('API', variant='secondary')
     saved_input = gr.State()
 
     with gr.Accordion(label='Advanced options', open=False):
@@ -165,6 +166,20 @@ with gr.Blocks(css='style.css') as demo:
 
     gr.Markdown(LICENSE)
 
+    api_button.submit(
+        fn=generate,
+        inputs=[
+            message,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+        ],
+        outputs=chatbot,
+        api_name="llama_api",
+    )
+    
     textbox.submit(
         fn=clear_and_save_textbox,
         inputs=textbox,