Upload folder using huggingface_hub

miner.py

@@ -142,7 +142,7 @@ class Miner:
         # self._warmup_models(device)
         
         # Increase batch sizes for better GPU utilization
-        self.player_batch_size = 64  # Increased from 32
+        self.player_batch_size = 8  # Increased from 32
         self.pitch_batch_size = 8   # Increased from 32
         print(f"✅ Keypoints Model Loaded")
 
@@ -155,6 +155,8 @@ class Miner:
         offset: int,
         n_keypoints: int,
     ) -> list[TVFrameResult]:
+
+        print("\n🤖 Running predict_batch...")
         player_batch_size = min(self.player_batch_size, len(batch_images))
         bboxes: dict[int, list[BoundingBox]] = {}
         while True:
@@ -198,6 +200,9 @@ class Miner:
                     player_batch_size = min(self.player_batch_size, len(batch_images))
                 else:
                     raise e
+            except Exception as e:
+                print(f"❌ Error during bbox prediction: {e}")
+                raise e
 
         pitch_batch_size = min(self.pitch_batch_size, len(batch_images))
         keypoints: dict[int, list[tuple[int, int]]] = {}
@@ -254,6 +259,9 @@ class Miner:
                     pitch_batch_size = min(self.pitch_batch_size, len(batch_images))
                 else:
                     raise e
+            except Exception as e:
+                print(f"❌ Error during keypoints prediction: {e}")
+                raise e
 
         # Combine results
         results: list[TVFrameResult] = []