unist works, but video only 3 frames

module_microast/microast_model.py

@@ -1,11 +1,10 @@
 import torch
 
-from . import net_microAST
 import torch.nn.functional as F
 from torchvision import transforms
 
 
-def microast_inference(
+def inference_microast(
     src_img: torch.Tensor,
     style_img: torch.Tensor,
     size: int,