Refactored according to linting style

app.py

@@ -1,20 +1,35 @@
-from neukit.gui import WebUI
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
+
+from neukit.gui import WebUI
 
 
 def main():
     parser = ArgumentParser()
-    parser.add_argument("--cwd", type=str, default="/home/user/app/", help="Set current working directory (path to app.py).")
-    parser.add_argument("--share", type=int, default=1, help="Whether to enable the app to be accessible online -> setups a public link which requires internet access.")
+    parser.add_argument(
+        "--cwd",
+        type=str,
+        default="/home/user/app/",
+        help="Set current working directory (path to app.py).",
+    )
+    parser.add_argument(
+        "--share",
+        type=int,
+        default=1,
+        help="Whether to enable the app to be accessible online"
+        "-> setups a public link which requires internet access.",
+    )
     args = parser.parse_args()
 
     print("Current working directory:", args.cwd)
 
     if not os.path.exists(args.cwd):
         raise ValueError("Chosen 'cwd' is not a valid path!")
-    if not args.share in [0, 1]:
-        raise ValueError("The 'share' argument can only be set to 0 or 1 (boolean), but was:", args.share)
+    if args.share not in [0, 1]:
+        raise ValueError(
+            "The 'share' argument can only be set to 0 or 1, but was:",
+            args.share,
+        )
 
     # initialize and run app
     print("Launching demo...")

neukit/gui.py

@@ -1,6 +1,7 @@
-import gradio as gr
 import os
 
+import gradio as gr
+
 from .inference import run_model
 from .utils import load_ct_to_numpy
 from .utils import load_pred_volume_to_numpy
@@ -8,7 +9,12 @@ from .utils import nifti_to_glb
 
 
 class WebUI:
-    def __init__(self, model_name: str = None, cwd: str = "/home/user/app/", share: int = 1):
+    def __init__(
+        self,
+        model_name: str = None,
+        cwd: str = "/home/user/app/",
+        share: int = 1,
+    ):
         # global states
         self.images = []
         self.pred_images = []
@@ -166,4 +172,6 @@ class WebUI:
         # https://gradio.app/sharing-your-app/
         # inference times > 60 seconds -> need queue():
         # https://github.com/tloen/alpaca-lora/issues/60#issuecomment-1510006062
-        demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=self.share)
+        demo.queue().launch(
+            server_name="0.0.0.0", server_port=7860, share=self.share
+        )