SerdarHelli commited on
Commit
d997382
1 Parent(s): 62b84f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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="/content/"
 
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':