Spaces:
Running
Running
Ayushnangia
commited on
Commit
•
e84bd0d
1
Parent(s):
8806fc4
Update raft.py
Browse files
raft.py
CHANGED
@@ -99,7 +99,7 @@ if __name__ == '__main__':
|
|
99 |
from imread_from_url import imread_from_url
|
100 |
|
101 |
# Initialize model
|
102 |
-
model_path='
|
103 |
flow_estimator = Raft(model_path)
|
104 |
|
105 |
# Read inference image
|
|
|
99 |
from imread_from_url import imread_from_url
|
100 |
|
101 |
# Initialize model
|
102 |
+
model_path='raft_things_iter20_480x640.onnx'
|
103 |
flow_estimator = Raft(model_path)
|
104 |
|
105 |
# Read inference image
|