update app

app.py

@@ -18,9 +18,10 @@ def predict_fluency(text):
     probs = torch.softmax(logits, dim=1)
     label = torch.argmax(probs, dim=1).item()
     if label == 0:
-        return "Low Fluency"
+        label = "Low Fluency"
     else:
-        return "High Fluency"
+        label = "High Fluency"
+    return f"{label}\nLow Fluency: {probs[0][0].item()}\nHigh Fluency: {probs[0][1].item()}"
 
 iface = gr.Interface(fn=predict_fluency, 
                      inputs="text", 
@@ -29,6 +30,8 @@ iface = gr.Interface(fn=predict_fluency,
                            examples=[
                                 ["Engineer, Yeah, indeed, you know that the lady has a Phd. It's the 1st.", 
                                  "Engineer, Yeah, indeed, you know that the lady has a Phd. It's the 1st."],
+                                ["Engineer, indeed, the lady, an accomplished engineer, holds a prestigious Ph.D. It's her first achievement of such caliber.",
+                                 "Engineer, indeed, the lady, an accomplished engineer, holds a prestigious Ph.D. It's her first achievement of such caliber."],
                                 [ "Oh, how was brown for you?", 
                                  "Oh, how was brown for you?"],
                                 [ "Now they can.", 
@@ -36,7 +39,7 @@ iface = gr.Interface(fn=predict_fluency,
                                 [ "But kind of plastics like growing more social consciousness, right?", 
                                   "But kind of plastics like growing more social consciousness, right?"]
                                       ]
-                     )
+                     
 
 
 iface.launch()