Update colorization.py

colorization.py

@@ -57,15 +57,15 @@ def inferenceColorize(img, version, scale):
         else:
             extension = 'jpg'
         '''
-        os.makedirs('output', exist_ok=True)
-        model.predict(img.name)
-        save_path = f'output/DeOldify/'+Path(img.name).stem+'.{extension}'
+    os.makedirs('output', exist_ok=True)
+    model.predict(img.name)
+    save_path = f'output/DeOldify/'+Path(img.name).stem+'.{extension}'
     
         '''
         cv2.imwrite(save_path, output)
         output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
         '''
-        return img, save_path
+    return img, save_path
     '''
     except Exception as error:
         print('global exception', error)