Support Google Colab

Mrz1526 · Apr 26, 2024 · 6b98966 · 6b98966
1 parent 8267587
commit 6b98966
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -218,6 +218,7 @@ def create_example(input_list):
     parser.add_argument('--share', action='store_true')
     parser.add_argument('--local-debug', action='store_true')
     parser.add_argument('--save-path', default='samples')
+    parser.add_argument('--share', action='store_true', default=False)
 
     args, unknownargs = parser.parse_known_args()
     LOCAL_DEBUG = args.local_debug
@@ -231,4 +232,4 @@ def create_example(input_list):
     demo.launch(server_name=args.server_name,
                 server_port=args.port, max_threads=40,
                 allowed_paths=['example/barbie2.jpg'],
-                share=True)
+                share=args.share)
diff --git a/colab.ipynb b/colab.ipynb
@@ -27,7 +27,7 @@
     "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://cloudbook-public-production.oss-cn-shanghai.aliyuncs.com/animation/animate_anything_512_v1.02.tar -d output/latent\n",
     "!tar -xf output/latent/animate_anything_512_v1.02.tar -C output/latent/\n",
     "\n",
-    "!python app.py --config output/latent/animate_anything_512_v1.02/config.yaml"
+    "!python app.py --config output/latent/animate_anything_512_v1.02/config.yaml --share"
    ],
    "metadata": {
     "collapsed": false,