Charbel Malo commited on
Commit
dbd6b54
·
verified ·
1 Parent(s): 6b091bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -35,12 +35,9 @@ parser = argparse.ArgumentParser(description="Swap-Mukham Face Swapper")
35
  parser.add_argument("--out_dir", help="Default Output directory", default=os.getcwd())
36
  parser.add_argument("--batch_size", help="Gpu batch size", default=32)
37
  parser.add_argument("--cuda", action="store_true", help="Enable cuda", default=False)
38
- parser.add_argument(
39
- "--colab", action="store_true", help="Enable colab mode", default=False
40
- )
41
-
42
  parser.add_argument("--device", default="cuda:0", type=str)
43
- args = parser.parse_args()
44
 
45
  @spaces.GPU
46
  def find_cuda():
 
35
  parser.add_argument("--out_dir", help="Default Output directory", default=os.getcwd())
36
  parser.add_argument("--batch_size", help="Gpu batch size", default=32)
37
  parser.add_argument("--cuda", action="store_true", help="Enable cuda", default=False)
38
+ parser.add_argument("--colab", action="store_true", help="Enable colab mode", default=False)
 
 
 
39
  parser.add_argument("--device", default="cuda:0", type=str)
40
+ user_args = parser.parse_args()
41
 
42
  @spaces.GPU
43
  def find_cuda():