Upload 2 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+172620-847860540_small.mp4 filter=lfs diff=lfs merge=lfs -text

172620-847860540_small.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06c3826b9b3f98a0172234b91f2a0a7bb579043e2617d5e33eed42997ab523da
+size 12098714

app.py

@@ -19,7 +19,6 @@ import utils.misc
 import utils.saveload
 from nets.blocks import InputPadder
 from nets.net34 import Net
-from tensorboardX import SummaryWriter
 import imageio
 from demo_dense_visualize import Tracker
 import spaces
@@ -55,6 +54,7 @@ state_dict = torch.hub.load_state_dict_from_url(url, map_location='cpu')
 model = Net(16)
 count_parameters(model)
 model.load_state_dict(state_dict, strict=True)
+print('loaded ckpt')
 model.cuda()
 for n, p in model.named_parameters():
     p.requires_grad = False
@@ -249,7 +249,7 @@ if __name__ == '__main__':
         
         with gr.Row():
             with gr.Column():
-                video_input = gr.Video(label="Upload Video", value="data/244754_medium.mp4")
+                video_input = gr.Video(label="Upload Video", value="172620-847860540_small.mp4")
                 extract_btn = gr.Button("Extract First Frame")
                 # Add sliders for resolution and sliding window length.
                 resolution_slider = gr.Slider(minimum=512, maximum=1024, step=256, value=1024, label="Target Resolution")