Upload folder using huggingface_hub

Dockerfile

@@ -14,4 +14,5 @@ RUN pip install --no-cache-dir -e .
 
 EXPOSE 8000
 
+ENV ENABLE_WEB_INTERFACE=true
 CMD ["uvicorn", "server.app:app", "--host", "0.0.0.0", "--port", "8000"]

server/wildfire_environment.py

@@ -12,6 +12,7 @@ This environment uses the FirenetCNN model for wildfire detection.
 
 import os
 import base64
+import numpy as np
 from uuid import uuid4
 from io import BytesIO
 
@@ -89,9 +90,12 @@ class WildfireEnvironment(Environment):
             gradcam_summary=f"Grad-CAM: {obs.get('gradcam_summary', [0])[0]}",
             frame_id=int(obs.get("frame_id", [0])[0]),
             step=int(obs.get("step", [0])[0]),
-            ground_truth=["fire", "smoke", "no_fire"][
-                int(obs.get("ground_truth", [2])[0])
-            ],
+            ground_truth_arr = obs.get("ground_truth", np.array([2]))
+            if hasattr(ground_truth_arr, 'ndim') and ground_truth_arr.ndim > 0:
+                gt_idx = int(ground_truth_arr[0])
+            else:
+                gt_idx = int(ground_truth_arr)
+            ground_truth=["fire", "smoke", "no_fire"][gt_idx],
             reward=reward,
             done=done,
             metadata={},