add share arg

app.py

@@ -1,6 +1,7 @@
 import json
 import os
 
+import argparse
 import numpy as np
 import torch
 import gradio as gr
@@ -110,4 +111,14 @@ with block:
     run_button.click(fn=process, inputs=ips, outputs=[generated_output, mask_output])
 
 
-block.launch()
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    '--share',
+    '-s',
+    action="store_true",
+    default=False,
+    help='Create public link for the app.'
+)
+args = parser.parse_args()
+
+block.launch(share=args.share)