support only single batch for now

main.py

@@ -212,20 +212,13 @@ def generate_strokes(
                 prime=primingData
             )
             
-            if len(stroke_tensors) == 1 and stroke_tensors[0].dim() == 2:
+            # batch_size is 1
+            if len(stroke_tensors) == 1:
                 all_strokes_tensor = stroke_tensors[0]
                 stroke_offsets = all_strokes_tensor.cpu().numpy().tolist()
             else:
                 stroke_offsets = []
-                for stroke_tensor in stroke_tensors:
-                    if stroke_tensor.dim() == 2:
-                        stroke_data = stroke_tensor.squeeze(0).cpu().numpy().tolist()
-                    else:
-                        stroke_data = stroke_tensor.cpu().numpy().tolist()
-                    
-                    if len(stroke_data) == 3:
-                        stroke_offsets.append(stroke_data)
-            
+                logger.warning(f"Expected single batch, but got {len(stroke_tensors)}")
             return stroke_offsets
             
         except Exception as e: