fix no such file error

segment_utils.py

@@ -1,6 +1,7 @@
 import numpy as np
 import mediapipe as mp
 import uuid
+import os
 
 from PIL import Image
 from mediapipe.tasks import python
@@ -28,6 +29,9 @@ def restore_result(croper, category, generated_image):
         extension = 'png'
     else:
         extension = 'jpg'
+    
+    if not os.path.exists("output"):
+        os.makedirs("output")
 
     path = f"output/{uuid.uuid4()}.{extension}"
     restored_image.save(path, quality=100)