Update app.py (#2)

- Update app.py (baaac941d352601fe7ca90116934fa8863462635)


Co-authored-by: Abubakar Abid <[email protected]>

app.py

@@ -8,6 +8,8 @@ import gradio as gr
 import requests
 from transformers import pipeline
 
+pipe = pipeline("sentiment-analysis")
+
 demo = gr.Blocks()
 
 with demo:
@@ -26,11 +28,12 @@ with demo:
     # Generate model prediction
     # Default model: distilbert-base-uncased-finetuned-sst-2-english
     def _predict(txt, tgt, state):
-        pipe = pipeline("sentiment-analysis")
         pred = pipe(txt)[0]
+        other_label = 'negative' if pred['label'].lower() == "positive" else "positive"
+        pred_confidences = {pred['label'].lower(): pred['score'], other_label: 1 - pred['score']}
 
         pred["label"] = pred["label"].title()
-        ret = f"Target: {tgt}. Model prediction: {pred['label']} ({pred['score']} confidence). {pred['label'] != tgt}\n\n"
+        ret = f"Target: **{tgt}**. Model prediction: **{pred['label']}**\n\n"
         if pred["label"] != tgt:
             state["fooled"] += 1
             ret += " You fooled the model! Well done!"
@@ -43,13 +46,14 @@ with demo:
         toggle_final_submit = gr.update(visible=done)
         toggle_example_submit = gr.update(visible=not done)
         new_state_md = f"State: {state['cnt']}/{total_cnt} ({state['fooled']} fooled)"
-        return ret, state, toggle_example_submit, toggle_final_submit, new_state_md
+        return pred_confidences, ret, state, toggle_example_submit, toggle_final_submit, new_state_md
 
     # Input fields
     text_input = gr.Textbox(placeholder="Enter model-fooling statement", show_label=False)
     labels = ["Positive", "Negative"]
     random.shuffle(labels)
     label_input = gr.Radio(choices=labels, label="Target (correct) label")
+    label_output = gr.Label()
     text_output = gr.Markdown()
     with gr.Column() as example_submit:
         submit_ex_button = gr.Button("Submit")
@@ -69,7 +73,7 @@ with demo:
     submit_ex_button.click(
         _predict,
         inputs=[text_input, label_input, state],
-        outputs=[text_output, state, example_submit, final_submit, state_display],
+        outputs=[label_output, text_output, state, example_submit, final_submit, state_display],
     )
 
     submit_hit_button.click(
@@ -79,4 +83,4 @@ with demo:
         _js="function(state, dummy) { return [state, window.location.search]; }",
     )
 
-demo.launch(favicon_path="https://huggingface.co/favicon.ico")
+demo.launch(favicon_path="https://huggingface.co/favicon.ico")