MetaQu commited on
Commit
d4d87dd
·
verified ·
1 Parent(s): 9bc11a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -43
app.py CHANGED
@@ -1,45 +1,14 @@
1
  from transformers import pipeline
2
- import gradio as gr
3
- from PIL import ExifTags
4
 
5
- # Pakai model publik (gratis, tanpa login)
6
- classifier = pipeline("image-classification", model="orpatashnik/image-real-fake")
7
-
8
- def detect_image(image):
9
- # --- Cek metadata (EXIF) ---
10
- exif_result = "⚠️ Tidak ada metadata kamera"
11
- try:
12
- raw_exif = image._getexif()
13
- if raw_exif:
14
- exif_result = "✅ Metadata kamera terdeteksi"
15
- except:
16
- pass
17
-
18
- # --- Prediksi model ---
19
- results = classifier(image)
20
- label = results[0]["label"]
21
- score = results[0]["score"] * 100
22
-
23
- if "fake" in label.lower():
24
- verdict = "🚨 Kemungkinan besar Hasil AI"
25
- else:
26
- verdict = "📷 Kemungkinan besar Foto Asli"
27
-
28
- return f"""{verdict}
29
-
30
- Label Model: {label}
31
- Confidence: {score:.2f}%
32
-
33
- Cek Metadata: {exif_result}
34
- """
35
-
36
- # Gradio app
37
- iface = gr.Interface(
38
- fn=detect_image,
39
- inputs=gr.Image(type="pil"),
40
- outputs="text",
41
- title="AI Image Detector (Gratis)",
42
- description="Upload gambar untuk mendeteksi apakah foto asli atau hasil AI"
43
- )
44
-
45
- iface.launch()
 
1
  from transformers import pipeline
 
 
2
 
3
+ # Menggunakan model publik dan gratis untuk klasifikasi gambar.
4
+ # Model 'orpatashnik/image-real-fake' diganti karena tidak dapat diakses.
5
+ classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
6
+
7
+ # Ini adalah contoh penggunaan, Anda bisa mengadaptasi sesuai kebutuhan.
8
+ # Contoh: Melakukan klasifikasi pada file gambar.
9
+ # from PIL import Image
10
+ # import requests
11
+ # url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/cat-dog.jpg"
12
+ # image = Image.open(requests.get(url, stream=True).raw)
13
+ # result = classifier(image)
14
+ # print(result)