Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ from torch.hub import download_url_to_file, get_dir
|
|
64 |
import shlex
|
65 |
import uuid
|
66 |
import mimetypes
|
67 |
-
|
68 |
|
69 |
os.environ["MASTER_ADDR"] = "127.0.0.1"
|
70 |
os.environ["MASTER_PORT"] = "12355"
|
|
|
64 |
import shlex
|
65 |
import uuid
|
66 |
import mimetypes
|
67 |
+
import torchvision.transforms as T
|
68 |
|
69 |
os.environ["MASTER_ADDR"] = "127.0.0.1"
|
70 |
os.environ["MASTER_PORT"] = "12355"
|