Update app.py

app.py

@@ -1,28 +1,26 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load model dari Hugging Face (pretrained AI image detector)
-classifier = pipeline("image-classification", model="dima806/fake-image-detection")
+# Pakai model yang masih tersedia (public)
+classifier = pipeline("image-classification", model="orpatashnik/StyleGAN-nada-detector")
 
-def detect_image(img):
-    results = classifier(img)
-    # Ambil hasil dengan skor tertinggi
-    top_result = max(results, key=lambda x: x["score"])
-    label = top_result["label"]
-    score = round(top_result["score"] * 100, 2)
-    
-    if label.lower() == "fake":
-        return f"🚨 Gambar kemungkinan besar AI Generated ({score}%)"
+def detect_ai(image):
+    results = classifier(image)
+    label = results[0]['label']
+    score = results[0]['score']
+    if "real" in label.lower():
+        return f"🌍 Foto Asli (confidence: {score:.2f})"
     else:
-        return f"✅ Gambar kemungkinan besar Asli ({score}%)"
+        return f"🤖 Gambar AI (confidence: {score:.2f})"
 
-# Gradio UI
+# Buat interface Gradio
 demo = gr.Interface(
-    fn=detect_image,
-    inputs=gr.Image(type="filepath"),
-    outputs="text",
+    fn=detect_ai,
+    inputs=gr.Image(type="pil", label="Upload Gambar"),
+    outputs=gr.Textbox(label="Hasil Deteksi"),
     title="AI Image Detector",
-    description="Upload foto untuk mendeteksi apakah gambar asli atau AI generated."
+    description="Upload gambar untuk mendeteksi apakah gambar asli atau buatan AI."
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()