save original images

app.py

@@ -1,11 +1,23 @@
 import gradio as gr
 from ultralytics import YOLO
+import os
+from datetime import datetime
 import cv2
 
 model = YOLO('head_yolov8s.pt')
 
 def head_detect(path):
+    if path is None:
+        return None
+    
     img = cv2.imread(path)
+    # Save the original image
+    if not os.path.exists('uploaded_images'):
+        os.makedirs('uploaded_images')
+    current_time = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f'uploaded_images/{current_time}.jpg'
+    cv2.imwrite(filename, img)
+
     output = model(source=img)
     res = output[0].cpu().numpy()