fix input of inference

app.py

@@ -20,9 +20,9 @@ import time
 
 from visual_util import predictions_to_glb
 from streamvggt.models.streamvggt import StreamVGGT
-from vggt.utils.load_fn import load_and_preprocess_images
-from vggt.utils.pose_enc import pose_encoding_to_extri_intri
-from vggt.utils.geometry import unproject_depth_map_to_point_map
+from streamvggt.utils.load_fn import load_and_preprocess_images
+from streamvggt.utils.pose_enc import pose_encoding_to_extri_intri
+from streamvggt.utils.geometry import unproject_depth_map_to_point_map
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -73,8 +73,9 @@ def run_model(target_dir, model) -> dict:
 
     frames = []
     for i in range(images.shape[0]):
+        image = images[i].unsqueeze(0) 
         frame = {
-            "img": images[i]  
+            "img": image
         }
         frames.append(frame)