Update app.py
Browse files
app.py
CHANGED
|
@@ -95,6 +95,7 @@ def load_video(video_path):
|
|
| 95 |
|
| 96 |
@spaces.GPU
|
| 97 |
def inference_video(video):
|
|
|
|
| 98 |
vid = load_video(video)
|
| 99 |
|
| 100 |
# The model expects inputs of shape: B x C x H x W
|
|
|
|
| 95 |
|
| 96 |
@spaces.GPU
|
| 97 |
def inference_video(video):
|
| 98 |
+
os.system('nvidia-smi')
|
| 99 |
vid = load_video(video)
|
| 100 |
|
| 101 |
# The model expects inputs of shape: B x C x H x W
|