Upload ZeroGPU/app.py with huggingface_hub

ZeroGPU/app.py

@@ -0,0 +1,95 @@
+import spaces
+import gradio as gr
+import torch
+import sys
+from pathlib import Path
+
+# Add repo root to path
+ROOT = Path(__file__).resolve().parent.parent
+sys.path.insert(0, str(ROOT))
+sys.path.insert(0, str(ROOT / "external" / "TripoSG"))
+sys.path.insert(0, str(ROOT / "external" / "MV-Adapter"))
+
+@spaces.GPU(duration=120)
+def generate_shape(image, seed=42, steps=30, guidance=7.5):
+    from app import load_triposg, generate_shape as _gen
+    import numpy as np
+    from PIL import Image as PILImage
+
+    if image is None:
+        return None, "No image provided"
+
+    img_array = np.array(PILImage.fromarray(image).convert("RGB"))
+    glb_path, status = _gen(
+        input_image=img_array,
+        remove_background=True,
+        num_steps=steps,
+        guidance_scale=guidance,
+        seed=int(seed),
+        face_count=120000,
+    )
+    return glb_path, status
+
+
+@spaces.GPU(duration=300)
+def apply_texture(glb_path, image, seed=42):
+    from app import apply_texture as _tex
+    import numpy as np
+    from PIL import Image as PILImage
+
+    if glb_path is None or image is None:
+        return None, None, "Run shape generation first"
+
+    img_array = np.array(PILImage.fromarray(image).convert("RGB"))
+    textured_glb, mv_img, status = _tex(
+        glb_path=glb_path,
+        input_image=img_array,
+        remove_background=True,
+        variant="sdxl",
+        tex_seed=int(seed),
+        enhance_face=True,
+        rembg_threshold=0.5,
+        rembg_erode=2,
+    )
+    return textured_glb, mv_img, status
+
+
+with gr.Blocks(title="MeshForge — Portrait to 3D") as demo:
+    gr.Markdown("# MeshForge\n### Portrait → 3D Mesh with Texture")
+
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="Portrait", type="numpy")
+            seed = gr.Slider(0, 2**31 - 1, value=42, step=1, label="Seed")
+            steps = gr.Slider(10, 50, value=30, step=1, label="Shape Steps")
+            guidance = gr.Slider(1.0, 15.0, value=7.5, step=0.5, label="Guidance Scale")
+            shape_btn = gr.Button("1. Generate Shape", variant="primary")
+
+        with gr.Column():
+            shape_output = gr.Model3D(label="Shape")
+            shape_status = gr.Textbox(label="Status")
+
+    with gr.Row():
+        with gr.Column():
+            tex_seed = gr.Slider(0, 2**31 - 1, value=42, step=1, label="Texture Seed")
+            tex_btn = gr.Button("2. Apply Texture", variant="primary")
+        with gr.Column():
+            textured_output = gr.Model3D(label="Textured")
+            mv_preview = gr.Image(label="Multiview Preview")
+            tex_status = gr.Textbox(label="Status")
+
+    glb_state = gr.State(None)
+
+    shape_btn.click(
+        generate_shape,
+        inputs=[input_image, seed, steps, guidance],
+        outputs=[glb_state, shape_status],
+    ).then(lambda x: x, inputs=glb_state, outputs=shape_output)
+
+    tex_btn.click(
+        apply_texture,
+        inputs=[glb_state, input_image, tex_seed],
+        outputs=[textured_output, mv_preview, tex_status],
+    )
+
+demo.launch()