Upload folder using huggingface_hub

app.py

@@ -372,7 +372,7 @@ def demo_lam(flametracking, lam, cfg):
     def core_fn_space(image_path: str, video_params, working_dir):
         return core_fn(image_path, video_params, working_dir)
 
-    with gr.Blocks(analytics_enabled=False) as demo:
+    with gr.Blocks(analytics_enabled=False, delete_cache=[3600, 3600]) as demo:
 
         logo_url = './assets/images/logo.jpeg'
         logo_base64 = get_image_base64(logo_url)

flame_tracking_single_image.py

@@ -140,7 +140,9 @@ class FlameTrackingSingleImage:
         frame_index = 0
 
         # Bounding box detection
-        frame = torchvision.io.read_image(input_image_path)
+        # frame = torchvision.io.read_image(input_image_path)
+        frame = cv2.imread(input_image_path)[:, :, ::-1].copy()
+        frame = torch.Tensor(frame).permute(2, 0, 1).contiguous()[:3, ...]
         try:
             _, frame_bbox, _ = self.vgghead_encoder(frame, frame_index)
         except Exception:

lam/runners/__init__.py

@@ -17,5 +17,4 @@ from lam.utils.registry import Registry
 
 REGISTRY_RUNNERS = Registry()
 
-from .train import *
 from .infer import *