gQIR

Runtime error

aRy4n commited on Mar 6

Commit

9b15315

verified ·

1 Parent(s): b4fecc8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -504,7 +504,7 @@ class BurstColorPipeline:
             dropout = False
         raft_model = RAFT(RAFTArgs())
-        raft_path = APP_ROOT / "pretrained_ckpts" / "models" / "raft-things.pth"
         raft_dict = torch.load(str(raft_path), map_location="cpu")
         corrected = {}
         for k, v in raft_dict.items():

             dropout = False
         raft_model = RAFT(RAFTArgs())
+        raft_path =  "pretrained_ckpts/models/raft-things.pth"
         raft_dict = torch.load(str(raft_path), map_location="cpu")
         corrected = {}
         for k, v in raft_dict.items():