Added uncertainty

app.py

@@ -23,14 +23,14 @@ def pred_is_elon_musk(text):
     logis = results['logits'].detach()
     prob = F.softmax(logis, dim=1)[0][1]
     prediction = np.argmax(logis.numpy(), axis=1).flatten()
-    output1 = f"It's from Elon Musk with {prob*100: 0.2f}% probability." 
+    output1 = f"The predicted probability is {prob*100: 0.2f}%.\n"
     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:
         output2 = f"Therefore, maybe it is Not from Elon Musk."
-    return output1+output2
+    return output1 + output2
 
 iface = gr.Interface(pred_is_elon_musk, inputs="text", 
     outputs="text", title='“Is the tweet from Elon Musk?” Classifier',