Spaces:
Configuration error
Configuration 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 |
|