Update app.py

app.py

@@ -6,14 +6,7 @@ import numpy as np
 # ----------------------------
 # Inisialisasi model publik
 # ----------------------------
-
-# Model A: Deteksi wajah
-model_a = pipeline("image-classification", model="qualcomm/MediaPipe-Face-Detection")
-
-# Model B: Klasifikasi gambar umum
 model_b = pipeline("image-classification", model="microsoft/resnet-50")
-
-# Model C: Klasifikasi gambar umum
 model_c = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 # ----------------------------
@@ -22,13 +15,6 @@ model_c = pipeline("image-classification", model="google/vit-base-patch16-224")
 def detect_image(image):
     results = []
 
-    # Model A
-    res_a = model_a(image)[0]
-    label_a, score_a = res_a['label'].lower(), res_a['score']
-    # Sederhanakan: ada wajah = human, tidak = AI
-    label_a_final = 'human' if 'face' in label_a else 'ai'
-    results.append((label_a_final, score_a))
-
     # Model B
     res_b = model_b(image)[0]
     label_b, score_b = res_b['label'].lower(), res_b['score']
@@ -41,20 +27,12 @@ def detect_image(image):
     label_c_final = 'human' if 'person' in label_c or 'human' in label_c else 'ai'
     results.append((label_c_final, score_c))
 
-    # ----------------------------
-    # Voting + Threshold
-    # ----------------------------
+    # Voting + threshold
     votes = [r[0] for r in results]
-    scores = [r[1] for r in results]
-
-    # Mayoritas voting
     final_label = max(set(votes), key=votes.count)
-
-    # Rata-rata confidence untuk final label
     relevant_scores = [s for (l, s) in results if l == final_label]
-    avg_confidence = np.mean(relevant_scores) * 100  # ke persen
+    avg_confidence = np.mean(relevant_scores) * 100
 
-    # Tentukan hasil akhir
     if avg_confidence < 80:
         return "⚠️ Tidak Pasti (cek manual)", round(avg_confidence, 2)
     elif final_label == 'human':
@@ -70,7 +48,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=[gr.Textbox(label="Hasil Deteksi"), gr.Number(label="Confidence (%)")],
     title="AI vs Foto Asli Detector",
-    description="Unggah gambar, sistem akan mendeteksi apakah gambar asli atau dihasilkan AI menggunakan ensemble 3 model publik + voting."
+    description="Unggah gambar, sistem akan mendeteksi apakah gambar asli atau dihasilkan AI menggunakan ensemble 2 model publik + voting."
 )
 
 iface.launch()