Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ from torchvision.io import write_jpeg
|
|
32 |
import tempfile
|
33 |
from pathlib import Path
|
34 |
from urllib.request import urlretrieve
|
35 |
-
|
36 |
def write_flo(flow, filename):
|
37 |
"""
|
38 |
Write optical flow in Middlebury .flo format
|
|
|
32 |
import tempfile
|
33 |
from pathlib import Path
|
34 |
from urllib.request import urlretrieve
|
35 |
+
import tensorflow as tf
|
36 |
def write_flo(flow, filename):
|
37 |
"""
|
38 |
Write optical flow in Middlebury .flo format
|