Upload app.py

app.py

@@ -5,11 +5,8 @@ import argparse
 from wfd.webui import start_demo
 
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Demo script for WaveFunctionDiffusion.")
-    parser.add_argument(
-        "--colab",
-        action="store_true", 
-        help="Use it in Google Colab.",
-    )
-    args = parser.parse_args()
+    class Arguments:
+        colab = False
+        link_to_colab = True
+    args = Arguments()
     start_demo(args)