Changed app to use Block to add label names

app.py

@@ -20,4 +20,13 @@ def classify_pun(text):
     label = "Pun" if predictions[0][0][0] < predictions[0][0][1] else "Not a pun"
     return label, str(predictions[0][0])
 
-gr.Interface(fn=classify_pun, inputs=["text"], outputs=["text", "text"]).launch()
+#gr.Interface(fn=classify_pun, inputs=["text"], outputs=["text", "text"]).launch()
+with gr.Blocks() as demo:
+    text = gr.Textbox(label="Text")
+    output = gr.Textbox(label="Classification")
+    output2 = gr.Textbox(label="Probabilities")
+    greet_btn = gr.Button("Submit")
+    greet_btn.click(fn=classify_pun, inputs=name, outputs=[output, output2], api_name="classify_pun")
+
+
+demo.launch()