Add an optional flag to easily control whether Gradio is enabled

app.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 
 import os
+import argparse
 
 import gradio as gr
 import torch
@@ -27,4 +28,9 @@ with gr.Blocks(css="style.css") as demo:
             demo_stylization.render()
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--share", action="store_true", description="Share the app via Gradio")
+    args = parser.parse_args()
+
+    share = True if args.share else False
+    demo.queue(max_size=20).launch(share=share)