Update app.py

app.py

@@ -10,23 +10,21 @@ general = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 def detect_image(img):
     try:
-        # Prediksi dengan AI detector
         result1 = detector(img)
         label1 = result1[0]['label']
         conf1 = round(result1[0]['score'] * 100, 2)
 
-        # Prediksi dengan model general (untuk cek ganda)
         result2 = general(img)
         label2 = result2[0]['label']
         conf2 = round(result2[0]['score'] * 100, 2)
 
-        # Logika sederhana untuk memutuskan hasil
+        # Logika baru tanpa cek manual
         if "fake" in label1.lower() or "artificial" in label1.lower():
-            final = f"⚠️ Kemungkinan Besar AI Generated ({conf1}%)"
-        elif "real" in label1.lower():
-            final = f"✅ Kemungkinan Besar Foto Asli ({conf1}%)"
+            final = f"⚠️ Gambar ini kemungkinan besar dihasilkan AI ({conf1}%)"
+        elif "human" in label1.lower() and conf1 < 95:
+            final = f"⚠️ Gambar ini kemungkinan besar dihasilkan AI ({100-conf1:.2f}%)"
         else:
-            final = f"⚠️ Tidak Pasti (cek manual)"
+            final = f"✅ Gambar ini kemungkinan besar asli ({conf1}%)"
 
         output = f"""
 ### Hasil Deteksi: