Create app.py

app.py

@@ -0,0 +1,77 @@
+import gradio as gr
+import torch
+import numpy as np
+from PIL import Image
+from transformers import AutoImageProcessor, AutoModel
+from sklearn.metrics.pairwise import cosine_similarity
+
+# Load model pretrained (tanpa training ulang)
+processor = AutoImageProcessor.from_pretrained("facebook/dino-vits16")
+model = AutoModel.from_pretrained("facebook/dino-vits16")
+model.eval()
+
+# Simpan data sampling (memory sementara)
+database_features = []
+database_labels = []
+
+def extract_feature(image):
+    inputs = processor(images=image, return_tensors="pt")
+    with torch.no_grad():
+        outputs = model(**inputs)
+    feature = outputs.last_hidden_state.mean(dim=1).numpy()
+    return feature
+
+def tambah_sampling(image, label):
+    if image is None or label.strip() == "":
+        return "❌ Gambar dan label wajib diisi"
+
+    feature = extract_feature(image)
+    database_features.append(feature)
+    database_labels.append(label)
+
+    return f"✅ Sampling ditambahkan: {label}\nTotal data: {len(database_labels)}"
+
+def prediksi_gambar(image):
+    if len(database_features) == 0:
+        return "❌ Belum ada data sampling"
+
+    feature = extract_feature(image)
+    similarities = cosine_similarity(feature, np.vstack(database_features))
+    idx = np.argmax(similarities)
+
+    label_pred = database_labels[idx]
+    skor = similarities[0][idx]
+
+    return f"📌 HASIL PREDIKSI:\n\nGambar paling mirip dengan: **{label_pred}**\nSkor kemiripan: {skor:.2f}"
+
+with gr.Blocks(title="Supervised Tebak Gambar") as app:
+    gr.Markdown("## 🧠 Aplikasi Tebak Gambar (Supervised Learning)")
+    gr.Markdown(
+        "Upload gambar contoh (sampling) + label manual. "
+        "Kemudian upload gambar baru untuk diprediksi."
+    )
+
+    with gr.Tab("1️⃣ Tambah Data Sampling"):
+        img_sample = gr.Image(type="pil", label="Upload Gambar Sampling")
+        label_input = gr.Textbox(label="Label Gambar")
+        btn_add = gr.Button("Tambah ke Dataset")
+        output_add = gr.Textbox(label="Status")
+
+        btn_add.click(
+            tambah_sampling,
+            inputs=[img_sample, label_input],
+            outputs=output_add
+        )
+
+    with gr.Tab("2️⃣ Tebak Gambar"):
+        img_test = gr.Image(type="pil", label="Upload Gambar untuk Ditebak")
+        btn_pred = gr.Button("Tebak Gambar")
+        output_pred = gr.Textbox(label="Hasil Prediksi")
+
+        btn_pred.click(
+            prediksi_gambar,
+            inputs=img_test,
+            outputs=output_pred
+        )
+
+app.launch()