Update streamlit_app.py

streamlit_app.py

@@ -27,6 +27,7 @@ if module == "Text Misinformation":
         else:
             st.warning("Please enter some text.")
 
+
 # Image Deepfake Detection
 elif module == "Image Deepfake":
     st.title("🖼️ Deepfake Image Detection")
@@ -36,5 +37,16 @@ elif module == "Image Deepfake":
         st.image(image, caption="Uploaded Image", use_column_width=True)
         model = load_image_model()
         result = model(image)[0]
-        st.success(f"Prediction: {result['label']}")
+
+        # Map label_0 and label_1 to meaningful labels with icons and colors
+        label_map = {
+            "LABEL_0": ("Unauthentic", "❌", "red"),
+            "LABEL_1": ("Authentic", "✅", "green")
+        }
+        label, icon, color = label_map.get(result['label'].upper(), (result['label'], "❓", "gray"))
+
+        st.markdown(
+            f"<h4 style='color:{color};'>{icon} Prediction: {label}</h4>",
+            unsafe_allow_html=True
+        )
         st.write(f"Confidence: {result['score']:.2f}")