Spaces:
Runtime error
Runtime error
SerdarHelli
commited on
Commit
•
d997382
1
Parent(s):
62b84f2
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import sys
|
2 |
import os
|
3 |
|
4 |
-
os.system("https://github.com/dunbar12138/pix2pix3D.git")
|
5 |
sys.path.append("pix2pix3D")
|
6 |
|
7 |
from typing import List, Optional, Tuple, Union
|
@@ -120,7 +120,8 @@ models={"seg2cat":network_cat
|
|
120 |
}
|
121 |
|
122 |
device='cuda' if torch.cuda.is_available() else 'cpu'
|
123 |
-
outdir="
|
|
|
124 |
class CPU_Unpickler(pickle.Unpickler):
|
125 |
def find_class(self, module, name):
|
126 |
if module == 'torch.storage' and name == '_load_from_bytes':
|
|
|
1 |
import sys
|
2 |
import os
|
3 |
|
4 |
+
os.system("git clone https://github.com/dunbar12138/pix2pix3D.git")
|
5 |
sys.path.append("pix2pix3D")
|
6 |
|
7 |
from typing import List, Optional, Tuple, Union
|
|
|
120 |
}
|
121 |
|
122 |
device='cuda' if torch.cuda.is_available() else 'cpu'
|
123 |
+
outdir="./"
|
124 |
+
|
125 |
class CPU_Unpickler(pickle.Unpickler):
|
126 |
def find_class(self, module, name):
|
127 |
if module == 'torch.storage' and name == '_load_from_bytes':
|