create subfolder in inference

app.py

@@ -49,7 +49,12 @@ def detect_img(img,model):
 
     # save the upload image to Inference
     current_image_name = "current_img.jpg"  
-    save_current_path = os.path.join("./Inference/images", current_image_name)
+    img_folder = "Inference/images"
+    try:
+        os.mkdir(img_folder)
+    except FileExistsError:
+        print(f"Folder already exists.")
+    save_current_path = os.path.join(img_folder, current_image_name)
     img = np.array(img)
     img = img[:,:,::-1] #convert color because np change it to bgr
     cv2.imwrite(save_current_path, img)
@@ -226,6 +231,11 @@ def detect_vid(vid,model):
 
     # save the upload video to Inference
     current_video_name = "current_vid.mp4"  
+    vid_folder = "Inference/videos"
+    try:
+        mkdir(vid_folder)
+    except FileExistsError:
+        print(f"Folder already exists.")
     save_current_path = os.path.join("./Inference/videos", current_video_name)
 
     frames = cv2.VideoCapture(vid)