Upload folder using huggingface_hub

app.py

@@ -449,59 +449,56 @@ def gen3dmesh(img_input0, img_input1, img_input2, img_input3, back_proj, smooth_
     return run_inference3d(img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter)
 
 
-@torch.no_grad()
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("# [SIGGRAPH'24] CharacterGen: Efficient 3D Character Generation from Single Images with Multi-View Pose Calibration")
-        with gr.Row():
-            with gr.Column(variant="panel"):
-                img_input = gr.Image(type="pil", label="Upload Image(without background)", image_mode="RGBA", width=768, height=512)
-                gr.Examples(
-                    label="Example Images",
-                    examples=glob.glob("./2D_Stage/material/examples/*.png"),
-                    inputs=[img_input]
-                )
-                with gr.Row():
-                    width_input = gr.Number(label="Width", value=512)
-                    height_input = gr.Number(label="Height", value=768)
-                    seed_input = gr.Number(label="Seed", value=2333)
-                    remove_bg = gr.Checkbox(label="Remove Background (with algorithm)", value=True)
-            with gr.Column(variant="panel"):
-                timestep = gr.Slider(minimum=10, maximum=70, step=1, value=40, label="Timesteps")
-                button1 = gr.Button(value="Generate 4 Views")
-                with gr.Row():
-                    img_input0 = gr.Image(type="pil", label="Back Image", image_mode="RGBA", width=256, height=384)
-                    img_input1 = gr.Image(type="pil", label="Front Image", image_mode="RGBA", width=256, height=384)
-                with gr.Row():
-                    img_input2 = gr.Image(type="pil", label="Right Image", image_mode="RGBA", width=256, height=384)
-                    img_input3 = gr.Image(type="pil", label="Left Image", image_mode="RGBA", width=256, height=384)
-            with gr.Column(variant="panel"):
-                smooth_iter = gr.Slider(minimum=0, maximum=10, step=1, value=5, label="Laplacian Smoothing Iterations")
-                with gr.Row():
-                    back_proj = gr.Checkbox(label="Back Projection")
-                    button2 = gr.Button(value="Generate 3D Mesh")
-                output_dir = gr.Textbox(label="Output Directory")
-                with gr.Row():
-                    with gr.Tab("GLB"):
-                        output_model_glb = gr.Model3D( label="Output Model (GLB Format)", height=512)
-                        gr.Markdown("Note: The model shown here has a darker appearance. Download to get correct results.")
-                    with gr.Tab("OBJ"):
-                        output_model_obj = gr.Model3D( label="Output Model (OBJ Format)")
-                        gr.Markdown("Note: The model shown here's texture is mapped to vertex. Download to get correct results.")
-        button1.click(
-            fn=gen4views,
-            inputs=[img_input, width_input, height_input, seed_input, timestep, remove_bg],
-            outputs=[img_input2, img_input0, img_input3, img_input1],
-            api_name="generate_views"
-        )
-        button2.click(
-            gen3dmesh,
-            inputs=[img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter],
-            outputs=[output_dir, output_model_obj, output_model_glb],
-            api_name="generate_3d_mesh"
-        )
-    demo.queue()
-    demo.launch()
 
-if __name__ == "__main__":
-    main()
+with gr.Blocks() as demo:
+    gr.Markdown("# [SIGGRAPH'24] CharacterGen: Efficient 3D Character Generation from Single Images with Multi-View Pose Calibration")
+    with gr.Row():
+        with gr.Column(variant="panel"):
+            img_input = gr.Image(type="pil", label="Upload Image(without background)", image_mode="RGBA", width=768, height=512)
+            gr.Examples(
+                label="Example Images",
+                examples=glob.glob("./2D_Stage/material/examples/*.png"),
+                inputs=[img_input]
+            )
+            with gr.Row():
+                width_input = gr.Number(label="Width", value=512)
+                height_input = gr.Number(label="Height", value=768)
+                seed_input = gr.Number(label="Seed", value=2333)
+                remove_bg = gr.Checkbox(label="Remove Background (with algorithm)", value=True)
+        with gr.Column(variant="panel"):
+            timestep = gr.Slider(minimum=10, maximum=70, step=1, value=40, label="Timesteps")
+            button1 = gr.Button(value="Generate 4 Views")
+            with gr.Row():
+                img_input0 = gr.Image(type="pil", label="Back Image", image_mode="RGBA", width=256, height=384)
+                img_input1 = gr.Image(type="pil", label="Front Image", image_mode="RGBA", width=256, height=384)
+            with gr.Row():
+                img_input2 = gr.Image(type="pil", label="Right Image", image_mode="RGBA", width=256, height=384)
+                img_input3 = gr.Image(type="pil", label="Left Image", image_mode="RGBA", width=256, height=384)
+        with gr.Column(variant="panel"):
+            smooth_iter = gr.Slider(minimum=0, maximum=10, step=1, value=5, label="Laplacian Smoothing Iterations")
+            with gr.Row():
+                back_proj = gr.Checkbox(label="Back Projection")
+                button2 = gr.Button(value="Generate 3D Mesh")
+            output_dir = gr.Textbox(label="Output Directory")
+            with gr.Row():
+                with gr.Tab("GLB"):
+                    output_model_glb = gr.Model3D( label="Output Model (GLB Format)", height=512)
+                    gr.Markdown("Note: The model shown here has a darker appearance. Download to get correct results.")
+                with gr.Tab("OBJ"):
+                    output_model_obj = gr.Model3D( label="Output Model (OBJ Format)")
+                    gr.Markdown("Note: The model shown here's texture is mapped to vertex. Download to get correct results.")
+    button1.click(
+        fn=gen4views,
+        inputs=[img_input, width_input, height_input, seed_input, timestep, remove_bg],
+        outputs=[img_input2, img_input0, img_input3, img_input1],
+        api_name="generate_views"
+    )
+    button2.click(
+        gen3dmesh,
+        inputs=[img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter],
+        outputs=[output_dir, output_model_obj, output_model_glb],
+        api_name="generate_3d_mesh"
+    )
+
+demo.queue()
+demo.launch()