Update app.py

app.py

@@ -73,10 +73,9 @@ def predict_image(img, image_model):
     img_array = np.expand_dims(img_array, axis=0)
     prediction = image_model.predict(img_array)
     confidence = float(prediction[0][0]) * 100
-    if confidence > 50:
-        return "AI", confidence
-    else:
-        return "REAL", 100 - confidence
+    label = "AI" if confidence > 50 else "REAL"
+    score = confidence if confidence > 50 else 100 - confidence
+    return label, score
     
 
 # # ---------------------------
@@ -163,12 +162,13 @@ with tab2:
         img = Image.open(uploaded_file).convert("RGB")
         st.image(img, caption="Uploaded Image", use_container_width=True)
 
-        confidence = predict_image(img, image_model)
+        label, confidence = predict_image(img, image_model)
 
-        if confidence > 50:
+        if label == "AI":
             st.error(f"⚠️ AI GENERATED IMAGE ({confidence:.2f}%)")
         else:
-            st.success(f"✅ REAL IMAGE ({100 - confidence:.2f}%)")
+            st.success(f"✅ REAL IMAGE ({confidence:.2f}%)")
+
         st.progress(int(confidence))
 
 # ---------------------------