final

index.py

@@ -21,17 +21,19 @@ def predict(image):
         dimension_dict = dict()
         face_dict = dict()
         image_dim = []
-        # try:
-        dim, face, img = get_patches.extract_patches(
-            image, dimension_dict, face_dict, image_dim, patch_path)
-        print("extract patches pass")
-        return dim, face, img
-        # except:
-        #     print("extract patches fail")
-        #     return None, None, None
+        try:
+            dim, face, img = get_patches.extract_patches(
+                image, dimension_dict, face_dict, image_dim, patch_path)
+            print("extract patches pass")
+            return dim, face, img
+        except:
+            print("extract patches fail")
+            return None, None, None
 
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-    os.mkdir(os.path.join(BASE_DIR, 'patches'))
+
+    if not os.path.exists(os.path.join(BASE_DIR, 'patches')):
+        os.mkdir(os.path.join(BASE_DIR, 'patches'))
 
     dim, face, img = extract_patches(image)