greet button

app.py

@@ -19,4 +19,16 @@ def zs(premise,hypothesis):
     entail_prob = round(probs[:,2].item() * 100,2)
     return contra_prob, neut_prob, entail_prob
 
-gr.Interface(fn=zs, inputs=["text", "text"], outputs="text").launch()
+# gr.Interface(fn=zs, inputs=["text", "text"], outputs=["text","text","text"]).launch()
+
+
+with gr.Blocks() as demo:
+    premise = gr.Textbox(label="Premise")
+    hypothesis = gr.Textbox(label="Hypothesis")
+    entailment = gr.Textbox(label="Entailment Probability")
+    contradiction = gr.Textbox(label="Contradiction Probability")
+    neutral = gr.Textbox(label="Neutral Probability")
+    greet_btn = gr.Button("Compute")
+    greet_btn.click(fn=zs, inputs=[premise,hypothesis], outputs=[contradiction,neutral,entailment])
+
+demo.launch()