Create app.py

app.py

@@ -1,7 +1,21 @@
-import gradio as gr
+from dust3r.demo import *
 
-def greet(name):
-    return "Hello " + name + "!!"
+if __name__ == '__main__':
+    parser = get_args_parser()
+    args = parser.parse_args()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+    if args.tmp_dir is not None:
+        tmp_path = args.tmp_dir
+        os.makedirs(tmp_path, exist_ok=True)
+        tempfile.tempdir = tmp_path
+
+    if args.server_name is not None:
+        server_name = args.server_name
+    else:
+        server_name = '0.0.0.0' if args.local_network else '127.0.0.1'
+
+    model = load_model(args.weights, args.device)
+    # dust3r will write the 3D model inside tmpdirname
+    with tempfile.TemporaryDirectory(suffix='dust3r_gradio_demo') as tmpdirname:
+        print('Outputing stuff in', tmpdirname)
+        main_demo(tmpdirname, model, args.devie, args.image_size, server_name, args.server_port)