fix tensor

flash_vstream/serve/demo.py

@@ -72,8 +72,8 @@ class Chat:
  sample_pos = all_pos
 
  patch_numpy = vreader.get_batch(sample_pos).asnumpy()
- patch_images = Image.fromarray(patch_numpy)
- return patch_images
+ print("patch_numpy", patch_numpy.shape)
+ return patch_numpy
 
  @torch.inference_mode()
  def generate(self, images_tensor, prompt, first_run, state):