Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ import torch
|
|
3 |
torch.jit.script = lambda f: f
|
4 |
|
5 |
from download import OMG_download
|
|
|
6 |
|
7 |
import sys
|
8 |
sys.path.append('./')
|
@@ -698,7 +699,7 @@ if __name__ == '__main__':
|
|
698 |
'height': height,
|
699 |
'width': width,
|
700 |
}
|
701 |
-
|
702 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
703 |
main(device, args.segment_type)
|
704 |
|
|
|
3 |
torch.jit.script = lambda f: f
|
4 |
|
5 |
from download import OMG_download
|
6 |
+
download = OMG_download()
|
7 |
|
8 |
import sys
|
9 |
sys.path.append('./')
|
|
|
699 |
'height': height,
|
700 |
'width': width,
|
701 |
}
|
702 |
+
|
703 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
704 |
main(device, args.segment_type)
|
705 |
|