From 941c24fd4084bb8da84979b76e15a023a140f26b Mon Sep 17 00:00:00 2001 From: Nikita Karaev Date: Fri, 5 Jan 2024 16:17:50 +0000 Subject: [PATCH] add meta copyright --- gradio_demo/app.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/gradio_demo/app.py b/gradio_demo/app.py index 11dfe1c..2c59374 100644 --- a/gradio_demo/app.py +++ b/gradio_demo/app.py @@ -1,3 +1,10 @@ +# Copyright (c) Meta Platforms, Inc. and affiliates. +# All rights reserved. + +# This source code is licensed under the license found in the +# LICENSE file in the root directory of this source tree. + + import os import torch import gradio as gr @@ -22,7 +29,12 @@ def cotracker_demo( model = model.cuda() load_video = load_video.cuda() - model(video_chunk=load_video, is_first_step=True, grid_size=grid_size) + model( + video_chunk=load_video, + is_first_step=True, + grid_size=grid_size, + grid_query_frame=grid_query_frame, + ) for ind in range(0, load_video.shape[1] - model.step, model.step): pred_tracks, pred_visibility = model( video_chunk=load_video[:, ind : ind + model.step * 2]