Spaces:
Running
on
T4
Running
on
T4
Commit
•
d54bbc0
1
Parent(s):
f4aa030
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,7 +79,7 @@ example_videos = gr.components.Dataset(components=[sample_vid], samples=[[path]
|
|
| 79 |
|
| 80 |
def main():
|
| 81 |
args = parse_args()
|
| 82 |
-
args.device = 'cuda'
|
| 83 |
print('*** Now using %s.'%(args.device))
|
| 84 |
model = Model(device=args.device)
|
| 85 |
|
|
|
|
| 79 |
|
| 80 |
def main():
|
| 81 |
args = parse_args()
|
| 82 |
+
args.device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 83 |
print('*** Now using %s.'%(args.device))
|
| 84 |
model = Model(device=args.device)
|
| 85 |
|