diff --git a/FlashVSR/examples/WanVSR/infer_flashvsr_tiny.py b/FlashVSR/examples/WanVSR/infer_flashvsr_tiny.py
index a06829d..7a317af 100644
--- a/FlashVSR/examples/WanVSR/infer_flashvsr_tiny.py
+++ b/FlashVSR/examples/WanVSR/infer_flashvsr_tiny.py
@@ -125,6 +125,10 @@ def prepare_input_tensor(path: str, scale: float = 4,fps=30, dtype=torch.bfloat1
 
     if isinstance(path,torch.Tensor):
         total,h0,w0,_ = path.shape
+        if total == 1:
+            print("got image,repeating to 25 frames")
+            path = path.repeat(25, 1, 1, 1) 
+            total=25
         sW, sH, tW, tH = compute_scaled_and_target_dims(w0, h0, scale=scale, multiple=128)
         pil_list=tensor2pillist(path)