Update flow/flow_utils.py

flow/flow_utils.py

@@ -12,7 +12,13 @@ sys.path.insert(0, gmflow_dir)
 from gmflow.gmflow import GMFlow  # noqa: E702 E402 F401
 from utils.utils import InputPadder  # noqa: E702 E402
 
+import huggingface_hub
+
+repo_name = 'Anonymous-sub/Rerender'
+
 global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
+gmflow_path = huggingface_hub.hf_hub_download(
+    repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='models')
 
 
 def coords_grid(b, h, w, homogeneous=False, device=None):