Update app.py

app.py

@@ -20,8 +20,12 @@ def predict_image(image_file, conf_threshold):
     print(f"Confidence threshold: {conf_threshold}")
 
     # Clear previous results if any
-    for f in os.listdir(output_dir):
-        os.remove(os.path.join(output_dir, f))
+    for item in os.listdir(output_dir):
+        item_path = os.path.join(output_dir, item)
+        if os.path.isfile(item_path):
+            os.remove(item_path)
+        elif os.path.isdir(item_path):
+            shutil.rmtree(item_path)
 
     results = model.predict(source=image_file, conf=conf_threshold, save=True, project=output_dir, name="run", exist_ok=True)
 
@@ -64,4 +68,4 @@ with gr.Blocks() as demo:
         image_button.click(predict_image, inputs=[image_input, image_conf_slider], outputs=image_output)
 
 # 4. Launch the Gradio interface
-demo.launch(share=True)
+demo.launch(share=True)