Upload 5 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+aloevera.jpeg filter=lfs diff=lfs merge=lfs -text
+nangkacempedak.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+import torch
+from torchvision import models, transforms
+from PIL import Image
+
+# ====================================================================
+# 1. DEFINISIKAN ARSITEKTUR MODEL
+# ====================================================================
+# Arsitektur harus SAMA PERSIS dengan saat Anda training.
+# Muat daftar nama kelas dari file
+with open('class_names.txt', 'r') as f:
+    class_names = [line.strip() for line in f.readlines()]
+num_classes = len(class_names)
+
+# Inisialisasi model VGG11_bn
+model = models.vgg11_bn(pretrained=False) # Tidak perlu pre-trained karena kita akan load bobot sendiri
+num_ftrs = model.classifier[6].in_features
+model.classifier[6] = torch.nn.Linear(num_ftrs, num_classes)
+
+# ====================================================================
+# 2. MUAT BOBOT MODEL (STATE DICTIONARY) DARI FILE .pth
+# ====================================================================
+# Pastikan model.pth ada di path yang benar
+# Hugging Face Spaces akan meletakkannya di direktori yang sama
+MODEL_PATH = 'herbal_vgg11_optimized.pth'
+model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device('cpu')))
+model.eval() # Set model ke mode evaluasi
+
+# ====================================================================
+# 3. DEFINISIKAN FUNGSI PREDIKSI
+# ====================================================================
+# Definisikan transformasi gambar (harus sama dengan transformasi validasi saat training)
+transform = transforms.Compose([
+    transforms.Resize((128, 128)),
+    transforms.ToTensor(),
+    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
+])
+
+def predict(image):
+    """Fungsi untuk memprediksi gambar input."""
+    # Proses gambar input
+    image = Image.fromarray(image.astype('uint8'), 'RGB')
+    image = transform(image).unsqueeze(0)
+
+    # Lakukan prediksi
+    with torch.no_grad():
+        outputs = model(image)
+        probabilities = torch.nn.functional.softmax(outputs[0], dim=0)
+        
+        # Buat dictionary kepercayaan untuk setiap kelas
+        confidences = {class_names[i]: float(probabilities[i]) for i in range(num_classes)}
+        
+    return confidences
+
+# ====================================================================
+# 4. BUAT DAN LUNCURKAN ANTARMUKA GRADIO
+# ====================================================================
+interface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(label="Unggah Gambar Tanaman Herbal"),
+    outputs=gr.Label(num_top_classes=3, label="Hasil Prediksi"),
+    title="🌿 Deteksi Tanaman Herbal",
+    description="Demo untuk model klasifikasi tanaman herbal menggunakan VGG11. Unggah gambar dan lihat prediksinya.",
+    examples=[
+        # Anda bisa menambahkan path ke gambar contoh di sini jika Anda mengunggahnya juga
+        ["nangkacempedak.jpg"],
+        ["aloevera.jpeg"]
+    ]
+)
+
+# Luncurkan aplikasi
+interface.launch()

class_names.txt

@@ -0,0 +1,100 @@
+Abelmoschus Esculentus (Okra)
+Acorus Calamus (Dlingo)
+Aloe Vera (Lidah Buaya)
+Alstonia Scholaris (Pulai)
+Amaranthus Spinosus (Bayam Duri)
+Andrographis Paniculata (Sambiloto)
+Annona Muricata (Sirsak)
+Annona Squamosa (Srikaya)
+Anredera Cordifolia (Binahong)
+Apium Graveolens (Seledri)
+Artocarpus Heterophyllus (Nangka Mini)
+Artocarpus Integer (Nangka Cempedak)
+Averrhoa Bilimbi (Belimbing Wuluh)
+Blumea Balsamifera (Sembung)
+Borreria Hispida (Gempur Batu)
+Caesalpinia Sappan L (Secang)
+Caladium Cicolor (Keladi)
+Calendula Officinalis (Marigold)
+Canangium Odoratum (Kenanga)
+Catharanthus Roseus (Tapak Dara)
+Celosia Cristata (Jengger Ayam)
+Centella Asiatica (Pegagan)
+Cestrum Nocturnum (Sedap Malam)
+Chrysopogon Zizanioides (Akar Wangi)
+Citrus Amblycarpa (Jeruk Limau)
+Clinalanthus Nutans (Dandang Gendis)
+Clitoria Ternatea (Kembang Telang)
+Crinum Asiaticum (Bakung Putih)
+Curcuma Domestica (Kunyit)
+Cyclea Barbata (Cincau Hijau)
+Cymbopogon Nardus (Serai)
+Derris Elliptica (Tuba)
+Desmodium Triquitrum (Daun Duduk)
+Dioscorea Hispida (Gadung)
+Eleutherine Americana (Bawang Dayak)
+Euodia Suaveolens (Zodia)
+Eupatorium Triplinerve (Prasman)
+Euphorbia Tirucalli (Patah Tulang)
+Euphoria Longan (Kelengkeng)
+Ficus Carica (Tin)
+Ficus Septica (Awar-Awar)
+Graptophyllum Pictum (Daun Ungu)
+Gynura Segetum (Daun Dewa)
+Hibiscus Rosa-sinensis (Kembang Sepatu)
+Hibiscus Sabdariffa (Rosela)
+Houttoynia Cordata (Amis-Amisan)
+Hydrocotyle Sibthorpioides (Semanggi Gunung)
+Impatiens Balsamina (Pacar Air)
+Isotoma Longiflora (Ki Tolod)
+Jasminum Sambac (Melati)
+Jatropa Multifida (Betadin)
+Kaempferia Galanga (Kencur)
+Melaleuca Leucadendra (Kayu Putih)
+Melia Azedarach (Mindi)
+Melissa officinalis (Lemon Balm)
+Michelia Alba (Kembang Kantil)
+Mirabilis Jalapa (Bunga Pukul Empat)
+Morinda Citrifolia (Mengkudu)
+Morus Alba (Murbei)
+Muraya Paniculata (Kemuning)
+Murraya Koenigii (Daun Kari)
+Nepeta Cataria (Catnip)
+Nothopanax Scutellarium (Mangkokan)
+Ocimum Americanum (Selasih)
+Ocimum Basilicum (Kemangi)
+Olea Europaea (Zaitun)
+Orthosiphon Spicatus (Kumis Kucing)
+Pandanus Amaryllifolius (Pandan)
+Parameria Laevigata (Kayu Rapet)
+Peperomia Pellucida (Suruhan)
+Phaleria Macrocarpa (Mahkota Dewa)
+Physalis Angulata (Ceplukan)
+Phytolacca Americana (Mrico Kepyar)
+Piper Betle (Sirih)
+Piper Sarmentosum (Cabean)
+Plectranthus Scutellarioides (Miana)
+Pleomele Angustifolia (Suji)
+Pluchea Indica (Beluntas)
+Plumbago Zeylanica (Daun Encok)
+Pogostemon Cablin (Nilam)
+Pouteria Caimito (Abiu)
+Prunus Domestica (Plum)
+Psidium Guajava (Jambu Biji)
+Pyrus Communis (Pir)
+Raulvolvia Serpentina (Pule Pandak)
+Ricinus Communis (Jarak)
+Rosmarinus Officinalis (Rosemary)
+Ruellia Tuberosa (Ceplikan)
+Ruta Angustifolia (Inggu)
+Selaginella Doederleinii (Paku Rane)
+Syzygium Polyanthum (Salam)
+Talinum Triangulare (Som Jawa)
+Tinospora Cordifolia (Brotowali)
+Tithonia Diversifolia (Daun Insulin)
+Typhonium Flagelliforme (Keladi Tikus)
+Vitex Trifolia (Legundi)
+Zephyranthes Candida (Lili Hujan Putih)
+Zingiber Officinale (Jahe)
+Zingiber Zerumbet (Lempuyang Gajah)
+Ziziphus mauritiana Lam. (Bidara)

requirements.txt

@@ -0,0 +1,4 @@
+torch
+torchvision
+gradio
+Pillow