Update app.py

app.py

@@ -87,18 +87,18 @@ class_names = ['battery', 'biological', 'brown-glass', 'cardboard',
 
 # Define bin colors for each class
 bin_colors = {
-    'battery': 'Merah (Red)',        # Limbah berbahaya
-    'biological': 'Cokelat (Brown)',  # Limbah organik
-    'brown-glass': 'Hijau (Green)',   # Gelas berwarna coklat
-    'cardboard': 'Kuning (Yellow)',    # Limbah daur ulang
-    'clothes': 'Biru (Blue)',         # Pakaian dan tekstil
-    'green-glass': 'Hijau (Green)',    # Gelas berwarna hijau
-    'metal': 'Kuning (Yellow)',        # Limbah daur ulang
-    'paper': 'Kuning (Yellow)',        # Limbah daur ulang
-    'plastic': 'Kuning (Yellow)',      # Limbah daur ulang
-    'shoes': 'Biru (Blue)',           # Pakaian dan tekstil
-    'trash': 'Hitam (Black)',         # Limbah umum
-    'white-glass': 'Putih (White)'    # Gelas berwarna putih
+    'battery': 'Merah (Red)',        # Limbah berbahaya (B3)
+    'biological': 'Hijau (Green)',   # Limbah organik
+    'brown-glass': 'Kuning (Yellow)',# Gelas berwarna coklat (anorganik/daur ulang)
+    'cardboard': 'Biru (Blue)',      # Kertas (daur ulang)
+    'clothes': 'Kuning (Yellow)',    # Pakaian (dimasukkan sebagai daur ulang)
+    'green-glass': 'Kuning (Yellow)',# Gelas berwarna hijau (anorganik/daur ulang)
+    'metal': 'Kuning (Yellow)',      # Logam (anorganik/daur ulang)
+    'paper': 'Biru (Blue)',          # Kertas (daur ulang)
+    'plastic': 'Kuning (Yellow)',    # Plastik (anorganik/daur ulang)
+    'shoes': 'Kuning (Yellow)',      # Sepatu (dimasukkan sebagai daur ulang)
+    'trash': 'Hitam (Black)',        # Limbah umum
+    'white-glass': 'Kuning (Yellow)' # Gelas berwarna putih (anorganik/daur ulang)
 }
 
 # Define the prediction function
@@ -123,7 +123,7 @@ iface = gr.Interface(
         gr.Textbox(label="Jenis Sampah"), 
         gr.Textbox(label="Tong Sampah yang Sesuai")  # 2 output with label
     ],
-    title="Klasifikasi Sampah dengan ResNet50",
+    title="Klasifikasi Sampah dengan ResNet50 v1",
     description="Unggah gambar sampah, dan model akan mengklasifikasikannya ke dalam salah satu dari 12 kategori bersama dengan warna tempat sampah yang sesuai."
 )