Update app.py

app.py

@@ -10,7 +10,7 @@ model1 = pipeline("image-classification", model="microsoft/resnet-50")
 model2 = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 # ----------------------------
-# Fungsi deteksi dengan ensemble
+# Fungsi deteksi dengan ensemble tanpa "Tidak Pasti"
 # ----------------------------
 def detect_image(image):
     results = []
@@ -28,24 +28,29 @@ def detect_image(image):
     results.append((label2_final, score2))
 
     # ----------------------------
-    # Voting + threshold
+    # Voting atau ambil confidence tertinggi
     # ----------------------------
     votes = [r[0] for r in results]
-    final_label = max(set(votes), key=votes.count)
-
-    # Rata-rata confidence untuk label final
-    relevant_scores = [s for (l, s) in results if l == final_label]
-    avg_confidence = np.mean(relevant_scores) * 100
+    if votes[0] == votes[1]:
+        final_label = votes[0]  # mayoritas sama
+        final_conf = np.mean([r[1] for r in results if r[0]==final_label])*100
+    else:
+        # ambil model dengan confidence tertinggi
+        if results[0][1] > results[1][1]:
+            final_label = results[0][0]
+            final_conf = results[0][1]*100
+        else:
+            final_label = results[1][0]
+            final_conf = results[1][1]*100
 
-    # Tentukan hasil akhir
-    if avg_confidence < 80:
-        final_result = "⚠️ Tidak Pasti (cek manual)"
-    elif final_label == 'human':
-        final_result = f"✅ Foto Asli ({avg_confidence:.2f}%)"
+    # ----------------------------
+    # Hasil akhir
+    # ----------------------------
+    if final_label == 'human':
+        final_result = f"✅ Foto Asli ({final_conf:.2f}%)"
     else:
-        final_result = f"🤖 AI Detected ({avg_confidence:.2f}%)"
+        final_result = f"🤖 AI Detected ({final_conf:.2f}%)"
 
-    # Detail per model
     output = f"""
 ### Hasil Deteksi:
 {final_result}
@@ -63,7 +68,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs="markdown",
     title="AI vs Foto Asli Detector",
-    description="Unggah gambar, sistem akan mendeteksi apakah gambar kemungkinan besar asli atau dihasilkan AI menggunakan ensemble 2 model publik + voting."
+    description="Unggah gambar, sistem akan mendeteksi apakah gambar AI atau Foto Asli berdasarkan 2 model publik."
 )
 
 if __name__ == "__main__":