Charbel Malo
commited on
Update app.py
Browse files
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 |
-
|
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():
|