Update app.py

app.py

@@ -1,23 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# load pipeline deteksi AI vs Real
-pipe = pipeline("image-classification", model="microsoft/resnet-50")  
-# ⚠️ ganti dengan model pendeteksi AI/real sesuai yang kamu pakai
-# contoh lain: "microsoft/ai-image-detector"
+# gunakan model deteksi AI vs Real
+pipe = pipeline("image-classification", model="umm-maybe/AI-image-detector")
 
 def detect_ai(image):
     results = pipe(image)
-    # Ambil hasil prediksi teratas
     results = sorted(results, key=lambda x: x['score'], reverse=True)
     label = results[0]['label']
     confidence = results[0]['score']
 
-    # Ubah jadi lebih informatif
-    if "AI" in label or "Fake" in label:
-        status = "Kemungkinan besar AI"
+    if "ai" in label.lower() or "fake" in label.lower():
+        status = "⚠️ Kemungkinan besar AI Generated"
     else:
-        status = "Kemungkinan besar Foto Asli"
+        status = "✅ Kemungkinan besar Foto Asli"
 
     return f"{status}\n\nLabel: {label}\nConfidence: {confidence*100:.2f}%"