Update app.py

app.py

@@ -5,20 +5,14 @@ import numpy as np
 import cv2
 
 # ----------------------------
-# MODEL
+# MODEL DETEKSI AI
 # ----------------------------
 try:
-    hf_detector = pipeline("image-classification", model="umm-maybe/AI-image-detector")
+    hf_detector = pipeline("image-classification", model="ckpt/real-or-ai")  # model gratis HF khusus AI detection
 except Exception as e:
     hf_detector = None
     print("HF AI-detector gagal dimuat:", e)
 
-try:
-    general_model = pipeline("image-classification", model="google/vit-base-patch16-224")
-except Exception as e:
-    general_model = None
-    print("General classifier gagal dimuat:", e)
-
 # ----------------------------
 # ANALISIS LOKAL
 # ----------------------------
@@ -44,12 +38,11 @@ def has_camera_exif(image):
     return False
 
 # ----------------------------
-# DETEKSI HYBRID DENGAN THRESHOLD LEBIH RENDAH
+# DETEKSI HYBRID
 # ----------------------------
 def detect_image(image):
     output_lines = []
 
-    # -------- HF AI-detector --------
     hf_score = 0
     hf_label = "N/A"
     hf_conf = 0
@@ -58,47 +51,32 @@ def detect_image(image):
             result = hf_detector(image)
             hf_label = result[0]['label']
             hf_conf = result[0]['score'] * 100
-            if any(x in hf_label.lower() for x in ["fake", "ai", "artificial"]):
+            if "ai" in hf_label.lower() or "synthetic" in hf_label.lower():
                 hf_score = hf_conf
         except:
             hf_score = 0
 
-    # Jika HF confidence > 50 → langsung AI
-    if hf_score > 50:
+    # Threshold HF lebih rendah agar sensitif
+    if hf_score > 30:
         final_result = "🤖 AI Detected"
         weighted_score = hf_score
         output_lines.append(f"### Hasil Deteksi:\n{final_result}")
         output_lines.append(f"HF AI-detector: {hf_label} ({hf_conf:.2f}%)")
         return "\n".join(output_lines)
 
-    # -------- General model --------
-    general_score = 0
-    general_label = "N/A"
-    general_conf = 0
-    if general_model:
-        try:
-            result2 = general_model(image)
-            general_label = result2[0]['label']
-            general_conf = result2[0]['score'] * 100
-            if any(x in general_label.lower() for x in ["anime","cartoon","illustration","maya","3d"]):
-                general_score = general_conf
-        except:
-            general_score = 0
-
-    # -------- Analisis lokal --------
+    # Analisis lokal tambahan
     blur_score = calculate_blur(image)
     noise_score = calculate_noise(image)
     exif_present = has_camera_exif(image)
     local_score = 0
     if blur_score < 100 or noise_score < 10:
-        local_score += 50
+        local_score += 30
     if not exif_present:
-        local_score += 10
+        local_score += 20
 
-    # -------- Weighted Score (general + local) --------
-    weighted_score = general_score*0.7 + local_score*0.3
+    weighted_score = hf_score * 0.6 + local_score * 0.4
 
-    if weighted_score > 50:
+    if weighted_score > 40:
         final_result = "🤖 AI Detected"
     else:
         final_result = "✅ Foto Asli"
@@ -107,7 +85,6 @@ def detect_image(image):
     output_lines.append(f"### Hasil Deteksi:\n{final_result}")
     output_lines.append(f"Weighted Skor: {weighted_score:.2f}")
     output_lines.append(f"HF AI-detector: {hf_label} ({hf_conf:.2f}%)")
-    output_lines.append(f"General Model: {general_label} ({general_conf:.2f}%)")
     output_lines.append(f"Blur Score: {blur_score:.2f}")
     output_lines.append(f"Noise Score: {noise_score:.2f}")
     output_lines.append(f"Metadata Kamera: {'Ada' if exif_present else 'Tidak Ada'}")