added text box for the probabilities as output

app.py

@@ -20,6 +20,6 @@ def classify_pun(text):
     inputs = [tokenizer(text, truncation=True)]
     predictions = trainer.predict(inputs)
     label = "Pun" if predictions[0][0][0] < predictions[0][0][1] else "Not a pun"
-    return label
+    return label, str(predictions[0][0])
 
-gr.Interface(fn=classify_pun, inputs=["text"], outputs=["text"]).launch()
+gr.Interface(fn=classify_pun, inputs=["text"], outputs=["text", "text"]).launch()