Update app.py

app.py

@@ -1,25 +1,28 @@
 import gradio as gr
 from transformers import pipeline
 
-# Pakai model yang masih tersedia (public)
-classifier = pipeline("image-classification", model="orpatashnik/StyleGAN-nada-detector")
+# Gunakan model publik yang valid
+classifier = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")
 
 def detect_ai(image):
+    # image dari Gradio
     results = classifier(image)
     label = results[0]['label']
     score = results[0]['score']
-    if "real" in label.lower():
-        return f"🌍 Foto Asli (confidence: {score:.2f})"
+    # format output
+    if "Real" in label:
+        return f"✅ Foto Asli (confidence: {score:.2f})"
     else:
-        return f"🤖 Gambar AI (confidence: {score:.2f})"
+        return f"🤖 Gambar AI / Fake (confidence: {score:.2f})"
 
-# Buat interface Gradio
+# Interface Gradio
 demo = gr.Interface(
     fn=detect_ai,
     inputs=gr.Image(type="pil", label="Upload Gambar"),
     outputs=gr.Textbox(label="Hasil Deteksi"),
-    title="AI Image Detector",
-    description="Upload gambar untuk mendeteksi apakah gambar asli atau buatan AI."
+    title="AI vs Real Image Detector",
+    description="Upload gambar untuk mengecek apakah foto asli atau buatan AI.",
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":