Update app.py

app.py

@@ -12,6 +12,8 @@ import subprocess
 import zipfile
 from PIL import Image
 import gradio as gr
+import requests
+from io import BytesIO
 
 # Setup Kaggle API
 kaggle_dir = os.path.expanduser("~/.kaggle")
@@ -85,18 +87,18 @@ class_names = ['battery', 'biological', 'brown-glass', 'cardboard',
 
 # Define bin colors and image paths
 bin_info = {
-    'battery': ('Merah (Red)', '/main/red_bin.png'),
-    'biological': ('Hijau (Green)', '/main/green_bin.png'),
-    'brown-glass': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'cardboard': ('Biru (Blue)', '/main/blue_bin.png'),
-    'clothes': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'green-glass': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'metal': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'paper': ('Biru (Blue)', '/main/blue_bin.png'),
-    'plastic': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'shoes': ('Kuning (Yellow)', '/main/yellow_bin.png'),
-    'trash': ('Abu-abu (Gray)', '/main/gray_bin.png'),
-    'white-glass': ('Kuning (Yellow)', '/main/yellow_bin.png')
+    'battery': ('Merah (Red)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/red_bin.png'),
+    'biological': ('Hijau (Green)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/green_bin.png'),
+    'brown-glass': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'cardboard': ('Biru (Blue)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/blue_bin.png'),
+    'clothes': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'green-glass': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'metal': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'paper': ('Biru (Blue)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/blue_bin.png'),
+    'plastic': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'shoes': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png'),
+    'trash': ('Abu-abu (Gray)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/gray_bin.png'),
+    'white-glass': ('Kuning (Yellow)', 'https://huggingface.co/spaces/kendrickfff/resnet50_garbage_classification_v1.2/resolve/main/yellow_bin.png')
 }
 
 # Define the prediction function