asset path and logging

app.py

@@ -28,16 +28,30 @@ import plotly.express as px
 from spatstatapp.inference import inference_large
 from spatstatapp.plotting import coco_to_detections
 from spatstatapp.tile_training_data import load_bboxes
+import os
 
 import gradio as gr
 import numpy as np
 from PIL import Image, ImageDraw
 import cv2
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # %%	load image and detections										 |
-model_path = Path("models/best.pt")
 
-data_dir=Path("img")
+def get_asset_path(relative_path):
+    if os.getenv('SPACE_ID'):
+        # Running on HF Spaces
+        return os.path.join(os.getcwd(), relative_path)
+    # Running locally
+    return relative_path
+
+model_path = get_asset_path(Path("models/best.pt"))
+
+data_dir=get_asset_path(Path("img"))
 data_dir.exists()
 
 train_images = list(data_dir.glob('shells.png'))
@@ -388,5 +402,12 @@ with gr.Blocks(
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            show_error=True,
+            debug=True,
+            share=True
+    )
+