Added uncertainty

app.py

@@ -21,13 +21,16 @@ def pred_is_elon_musk(text):
     results = model(input_ids=ids, token_type_ids=None,
                     attention_mask=masks)
     logis = results['logits'].detach()
-    prob = F.softmax(logis, dim=1)[0]
+    prob = F.softmax(logis, dim=1)[0][1]
     prediction = np.argmax(logis.numpy(), axis=1).flatten()
-    if prediction[0] == 1:
-        output = f"It's from Elon Musk with {prob[0]*100: 0.2f}% probability."
+    output1 = f"It's from Elon Musk with {prob*100: 0.2f}% probability." 
+    if 0.4 <= prob <= 0.6:
+        output2 = f"Therefore, maybe it's from Elon Musk or maybe not."
+    elif prediction[0] == 1:
+        output2 = f"Therefore, maybe it is from Elon Musk."
     else:
-        output = f"It's NOT from Elon Musk with {prob[1]*100: 0.2f}% probability."
-    return output
+        output2 = f"Therefore, maybe it is Not from Elon Musk."
+    return output1+output2
 
 iface = gr.Interface(pred_is_elon_musk, inputs="text", 
     outputs="text", title='“Is the tweet from Elon Musk?” Classifier',