Update app.py

app.py

@@ -1,39 +1,43 @@
+# app.py
 import gradio as gr
 from transformers import pipeline
-from PIL import Image
 
-# Model publik (tidak perlu token)
-model1 = pipeline("image-classification", model="umm-maybe/ai-image-detector")
-model2 = pipeline("image-classification", model="google/vit-base-patch16-224")
+# Muat pipeline deteksi AI-generated image (bisa ganti model sesuai kebutuhan)
+# Model gratis: "microsoft/resnet-50" hanya sebagai contoh, 
+# lebih baik gunakan model yang memang fine-tuned untuk AI vs Real
+detector = pipeline("image-classification", model="falconsai/nsfw_image_detection")
 
-def detect_ai(image):
-    img = image.convert("RGB").resize((224, 224))
+# Fungsi untuk memproses input gambar
+def detect_ai_image(image):
+    results = detector(image)
 
-    res1 = model1(img)[0]
-    res2 = model2(img)[0]
+    # Ambil prediksi dengan skor tertinggi
+    label = results[0]['label']
+    score = results[0]['score']
 
-    label1, conf1 = res1['label'], res1['score']
-    label2, conf2 = res2['label'], res2['score']
-
-    # Voting sederhana: kalau model1 bilang FAKE, lebih dipercaya
-    if "fake" in label1.lower() or "ai" in label1.lower():
-        verdict = "🚨 Kemungkinan besar AI Generated"
-    elif "real" in label1.lower():
-        verdict = "✅ Kemungkinan besar Foto Asli"
+    # Tentukan kategori berdasarkan threshold
+    if score > 0.85 and label.lower() in ["artificial", "ai", "fake"]:
+        verdict = "⚠️ Pasti AI-Generated"
+    elif score < 0.15 and label.lower() in ["real", "photo", "natural"]:
+        verdict = "✅ Pasti Foto Asli"
     else:
-        verdict = "⚠️ Tidak Pasti (cek manual)"
-
-    return f"""{verdict}
+        verdict = "❓ Tidak Pasti (perlu cek manual)"
 
-Model AI-detector: {label1} ({conf1*100:.2f}%)
-Model General (ViT): {label2} ({conf2*100:.2f}%)"""
+    return {
+        "Prediksi Label": label,
+        "Tingkat Keyakinan": f"{score*100:.2f}%",
+        "Kesimpulan": verdict
+    }
 
+# UI Gradio
 demo = gr.Interface(
-    fn=detect_ai,
+    fn=detect_ai_image,
     inputs=gr.Image(type="pil"),
-    outputs="text",
-    title="Deteksi Foto AI vs Asli (Gratis)",
-    description="Menggunakan model publik gratis Hugging Face."
+    outputs="json",
+    title="AI vs Real Image Detector",
+    description="Unggah foto untuk mendeteksi apakah itu buatan AI atau foto asli. "
+                "Gunakan threshold >85% untuk 'AI pasti' dan <15% untuk 'Asli pasti'."
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()