Update app.py

app.py

@@ -128,6 +128,7 @@ import gradio as gr
 import torch
 import cv2
 import os
+import shutil
 
 # Load the model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)