Move app to root

app/app.py → app.py

@@ -14,9 +14,9 @@ import sys
 import numpy as np
 from PIL import Image, ImageDraw
 import gradio as gr
-from tiff_utils import extract_physical_metadata
-from dl_inference import inference_fn
-from knn import knn, segment_image, bokeh_plot_knn, color_palette
+from app.tiff_utils import extract_physical_metadata
+from app.dl_inference import inference_fn
+from app.knn import knn, segment_image, bokeh_plot_knn, color_palette
 
 import tempfile
 import shutil
@@ -368,6 +368,5 @@ block.launch(
     share=False,
     show_error=True,
     server_name="0.0.0.0",
-    server_port=9003,
     enable_queue=True,
 )

app/dl_inference.py

@@ -10,6 +10,7 @@
 from functools import lru_cache
 import sys
 
+from .tiff_utils import tiff_to_png
 if ".." not in sys.path:
     sys.path.append("..")
 
@@ -23,7 +24,6 @@ from utils.constants import ModelArgs
 from utils.paths import MODELS_PATH, DATASET_PATH
 from atoms_detection.dl_detection import DLDetection
 from atoms_detection.evaluation import Evaluation
-from tiff_utils import tiff_to_png
 
 LOGS_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs")
 VOID_DS = os.path.join(DATASET_PATH, "void.csv")

app/knn.py

@@ -15,7 +15,7 @@ import matplotlib
 import numpy as np
 import os
 import seaborn as sns
-from logger import logger
+from .logger import logger
 
 
 matplotlib.use("agg")