Update app.py

app.py

@@ -1,29 +1,27 @@
-import gradio as gr
 from transformers import pipeline
-from PIL import Image
+import gradio as gr
 
-# Gunakan model khusus deteksi AI-generated photo
-detector = pipeline("image-classification", model="microsoft/ai-image-detector")
+# Ganti model dengan yang khusus AI detection
+classifier = pipeline("image-classification", model="microsoft/ai-image-detector")
 
-def detect(image):
-    results = detector(image)
-    results = sorted(results, key=lambda x: x['score'], reverse=True)
-    label = results[0]['label']
-    confidence = results[0]['score'] * 100
+def detect_image(image):
+    results = classifier(image)
+    label = results[0]["label"]
+    score = results[0]["score"] * 100
 
     if "fake" in label.lower() or "ai" in label.lower():
-        return f"⚠️ Kemungkinan besar Gambar AI\n\nLabel: {label}\nConfidence: {confidence:.2f}%"
+        verdict = "Kemungkinan besar Hasil AI"
     else:
-        return f"✅ Kemungkinan besar Foto Asli\n\nLabel: {label}\nConfidence: {confidence:.2f}%"
+        verdict = "Kemungkinan besar Foto Asli"
+
+    return f"{verdict}\n\nLabel: {label}\nConfidence: {score:.2f}%"
 
-# UI dengan Gradio
-app = gr.Interface(
-    fn=detect,
-    inputs=gr.Image(type="pil", label="Upload Foto"),
+iface = gr.Interface(
+    fn=detect_image,
+    inputs=gr.Image(type="pil"),
     outputs="text",
-    title="AI vs Foto Detector",
-    description="Unggah foto untuk mendeteksi apakah ini foto asli atau hasil AI-generated."
+    title="Deteksi Foto AI vs Asli",
+    description="Upload foto untuk mendeteksi apakah asli atau hasil AI."
 )
 
-if __name__ == "__main__":
-    app.launch()
+iface.launch()