update with better model support

.gitignore

@@ -6,4 +6,5 @@ flagged/
 *.mkv
 .DS_Store
 gradio_cached_examples/
-venv/
+venv/
+__pycache__

README.md

@@ -1,6 +1,6 @@
 ---
-title: Ripe And Unripe Tomatoes Detection
-emoji: 🍅
+title: Waste Detection
+emoji: 🗑️
 colorFrom: green
 colorTo: yellow
 sdk: gradio

app.py

@@ -13,21 +13,29 @@ class_colors = {cls: (random.randint(0, 255), random.randint(0, 255), random.ran
 
 # File URLs for sample images and video
 file_urls = [
-    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/AluCan1%2C000.jpg',
-    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/Glass847.jpg',
-    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/sample_waste.mp4',
+    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/AluCan1%2C000.jpg?download=true',
+    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/Glass847.jpg?download=true',
+    'https://huggingface.co/spaces/iamsuman/waste-detection/resolve/main/samples/sample_waste.mp4?download=true',
 ]
 
-# Function to download files
+# Function to download files (always overwrites existing ones)
 def download_file(url, save_name):
-    if not os.path.exists(save_name):
-        file = requests.get(url)
-        open(save_name, 'wb').write(file.content)
+    print(f"Downloading from: {url}")  # Log the URL
+    try:
+        response = requests.get(url, stream=True)
+        response.raise_for_status()  # Check for HTTP errors
+        with open(save_name, 'wb') as file:
+            for chunk in response.iter_content(1024):
+                file.write(chunk)
+        print(f"Downloaded and overwritten: {save_name}")
+    except requests.exceptions.RequestException as e:
+        print(f"Error downloading {url}: {e}")
 
 # Download images and video
 for i, url in enumerate(file_urls):
+    print(i, url)
     if 'mp4' in file_urls[i]:
-        download_file(file_urls[i], "video.mp4")
+        download_file(file_urls[i], f"video.mp4")
     else:
         download_file(file_urls[i], f"image_{i}.jpg")
 

best.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dd309b6c4ad5c32e348aa4187ec03e4c558b50099caf846fb4f21b21b85c4228
-size 6243235
+oid sha256:c6b887e7a54f4dda0ac67bbe0efcf30edf25b3cd356ffa97f58e289427b22d4c
+size 6233059

samples/tomatoes.mp4

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:605d5fb1a532865bd93c5cc08d92b9804d8987f6aedc66c88e9ca32ad9932fc1
-size 2537426