Updating UI and Some fixes

app.py

@@ -14,6 +14,11 @@ if st.button("Predict"):
         st.warning("Please enter some text to analyze.")
     else:
         prediction, confidence = predict_text(text_input)
-        label = "🧠 Human" if prediction == 1 else "🤖 AI"
-        st.success(f"**Prediction:** {label}")
-        st.info(f"**Confidence:** {confidence*100:.2f}%")
+        
+        if prediction == 1:
+            st.success("**Class**: 🤖 AI-Generated")
+            st.info(f"**Confidence**: {confidence * 100:.2f}% (AI)")
+        else:
+            st.success("**Class**: 🧠 Human-Written")
+            st.info(f"**Confidence**: {(1 - confidence) * 100:.2f}% (Human)")
+

predictor.py

@@ -44,5 +44,5 @@ def preprocess_text(text):
 def predict_text(text):
     X_sample = preprocess_text(text)
     prediction = model.predict(X_sample)[0]
-    confidence = model.predict_proba(X_sample)[0][1] 
+    confidence = model.predict_proba(X_sample)[:,1][0] 
     return prediction, confidence