removed neutral

app.py

@@ -18,7 +18,9 @@ def zs(premise,hypothesis):
     contra_prob = round(probs[:,0].item() * 100,2)
     # neut_prob = round(probs[:,1].item() * 100,2)
     entail_prob = round(probs[:,2].item() * 100,2)
-    return contra_prob, neut_prob, entail_prob
+    # return contra_prob, neut_prob, entail_prob
+    return contra_prob, entail_prob
+
 
 # gr.Interface(fn=zs, inputs=["text", "text"], outputs=["text","text","text"]).launch()
 
@@ -32,7 +34,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         entailment = gr.Textbox(label="Entailment Probability")
         contradiction = gr.Textbox(label="Contradiction Probability")
-        neutral = gr.Textbox(label="Neutral Probability")
-        greet_btn.click(fn=zs, inputs=[premise,hypothesis], outputs=[contradiction,neutral,entailment])
+        # neutral = gr.Textbox(label="Neutral Probability")
+        # greet_btn.click(fn=zs, inputs=[premise,hypothesis], outputs=[contradiction,neutral,entailment])
+        greet_btn.click(fn=zs, inputs=[premise,hypothesis], outputs=[contradiction,entailment])
+
 
 demo.launch()