Upload folder using huggingface_hub

server/app.py

@@ -36,11 +36,19 @@ except Exception as e:  # pragma: no cover
     ) from e
 
 try:
-    from ..models import TrafficLightAction, TrafficLightObservation
-    from .traffic_light_env_environment import TrafficLightEnvironment
-except ModuleNotFoundError:
-    from models import TrafficLightAction, TrafficLightObservation
-    from server.traffic_light_env_environment import TrafficLightEnvironment
+    from traffic_light_env.models import TrafficLightAction, TrafficLightObservation
+    from traffic_light_env.server.traffic_light_env_environment import TrafficLightEnvironment
+except ImportError:
+    try:
+        from ..models import TrafficLightAction, TrafficLightObservation
+        from .traffic_light_env_environment import TrafficLightEnvironment
+    except ImportError:
+        import sys
+        from pathlib import Path
+
+        sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+        from models import TrafficLightAction, TrafficLightObservation
+        from server.traffic_light_env_environment import TrafficLightEnvironment
 
 
 # Create the app with web interface and README integration