Update app.py

app.py

@@ -7,15 +7,12 @@ import cv2
 # ----------------------------
 # MODEL
 # ----------------------------
-
-# Model AI-detector (Hugging Face, public)
 try:
     hf_detector = pipeline("image-classification", model="umm-maybe/AI-image-detector")
 except Exception as e:
     hf_detector = None
     print("HF AI-detector gagal dimuat:", e)
 
-# Model general classifier (backup)
 try:
     general_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 except Exception as e:
@@ -47,10 +44,12 @@ def has_camera_exif(image):
     return False
 
 # ----------------------------
-# DETEKSI HYBRID
+# DETEKSI HYBRID WEIGHTED
 # ----------------------------
 def detect_image(image):
-    scores = []
+    hf_score = 0
+    general_score = 0
+    local_score = 0
 
     # -------- HF AI-detector --------
     if hf_detector:
@@ -58,14 +57,9 @@ def detect_image(image):
             result = hf_detector(image)
             label = result[0]['label'].lower()
             conf = result[0]['score'] * 100
-            if "fake" in label or "ai" in label or "artificial" in label:
-                scores.append(conf)  # skor AI
-            else:
-                scores.append(0)
+            hf_score = conf if any(x in label for x in ["fake", "ai", "artificial"]) else 0
         except:
-            scores.append(0)
-    else:
-        scores.append(0)
+            hf_score = 0
 
     # -------- General model --------
     if general_model:
@@ -73,32 +67,24 @@ def detect_image(image):
             result2 = general_model(image)
             label2 = result2[0]['label'].lower()
             conf2 = result2[0]['score'] * 100
-            # Logika sederhana: label aneh / tidak realistis dianggap AI
-            if any(x in label2 for x in ["abaya", "cartoon", "anime", "illustration"]):
-                scores.append(conf2)
-            else:
-                scores.append(0)
+            general_score = conf2 if any(x in label2 for x in ["anime","cartoon","illustration","maya"]) else 0
         except:
-            scores.append(0)
-    else:
-        scores.append(0)
+            general_score = 0
 
     # -------- Analisis lokal --------
     blur_score = calculate_blur(image)
     noise_score = calculate_noise(image)
     exif_present = has_camera_exif(image)
-
-    local_ai_score = 0
+    local_score = 0
     if blur_score < 100 or noise_score < 10:
-        local_ai_score += 50
+        local_score += 50
     if not exif_present:
-        local_ai_score += 10
-    scores.append(local_ai_score)
+        local_score += 10
 
-    # -------- Rata-rata skor & threshold --------
-    avg_score = sum(scores) / len(scores)
+    # -------- Weighted Score --------
+    final_score = hf_score*0.6 + general_score*0.25 + local_score*0.15
 
-    if avg_score > 60:
+    if final_score > 50:
         final_result = "🤖 AI Detected"
     else:
         final_result = "✅ Foto Asli"
@@ -107,9 +93,9 @@ def detect_image(image):
 ### Hasil Deteksi:
 {final_result}
 
-**Skor rata-rata AI:** {avg_score:.2f}
-**Model AI-detector:** {result[0]['label']} ({result[0]['score']*100:.2f}%)
-**Model General:** {result2[0]['label']} ({result2[0]['score']*100:.2f}%)
+**Weighted Skor:** {final_score:.2f}
+**HF AI-detector:** {result[0]['label']} ({result[0]['score']*100:.2f}%)
+**General Model:** {result2[0]['label']} ({result2[0]['score']*100:.2f}%)
 **Blur Score:** {blur_score:.2f}
 **Noise Score:** {noise_score:.2f}
 **Metadata Kamera:** {'Ada' if exif_present else 'Tidak Ada'}
@@ -123,7 +109,7 @@ iface = gr.Interface(
     fn=detect_image,
     inputs=gr.Image(type="pil"),
     outputs="markdown",
-    title="AI vs Foto Asli Detector (Hybrid + Voting)",
+    title="AI vs Foto Asli Detector (Weighted Hybrid)",
     description="Unggah gambar, sistem hybrid akan mendeteksi apakah gambar kemungkinan besar asli atau dihasilkan AI."
 )