Upload 3 files

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 4.36.1
+sdk_version: 4.0.2
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -196,75 +196,66 @@ def generate_3d(
     return output_path
 
 
-# Interface Gradio
-with gr.Blocks(title="TripoSG 3D Generator") as demo:
-    gr.Markdown("""
-    # 🎨 TripoSG 3D Generator
+# Interface Gradio - usando Interface simples para evitar bugs de schema
+def generate_wrapper(image, num_steps, guidance_scale, seed, remove_bg, output_format):
+    """Wrapper para a função generate_3d sem progress."""
+    if image is None:
+        raise gr.Error("Por favor, faça upload de uma imagem.")
+
+    # Remover fundo se necessário
+    if remove_bg:
+        image = remove_background(image)
+
+    # Redimensionar para 512x512
+    image = image.resize((512, 512), Image.LANCZOS)
+
+    # Gerar
+    generator = torch.Generator(device=DEVICE).manual_seed(int(seed))
+
+    with torch.no_grad():
+        outputs = pipe(
+            image=image,
+            generator=generator,
+            num_inference_steps=int(num_steps),
+            guidance_scale=float(guidance_scale),
+            use_flash_decoder=False,
+        )
+
+    # Salvar mesh
+    mesh = outputs.meshes[0]
+
+    with tempfile.NamedTemporaryFile(suffix=f".{output_format}", delete=False) as f:
+        output_path = f.name
+
+    if output_format == "glb":
+        mesh.export(output_path)
+    else:
+        mesh.export(output_path, file_type="stl")
+
+    return output_path
+
 
+demo = gr.Interface(
+    fn=generate_wrapper,
+    inputs=[
+        gr.Image(label="Imagem de Entrada", type="pil"),
+        gr.Slider(label="Passos de inferência", minimum=20, maximum=100, value=50, step=5),
+        gr.Slider(label="Guidance Scale", minimum=1.0, maximum=15.0, value=7.0, step=0.5),
+        gr.Number(label="Seed", value=42, precision=0),
+        gr.Checkbox(label="Remover fundo automaticamente", value=True),
+        gr.Radio(label="Formato de saída", choices=["glb", "stl"], value="glb"),
+    ],
+    outputs=gr.File(label="Modelo 3D Gerado"),
+    title="🎨 TripoSG 3D Generator",
+    description="""
     Gere modelos 3D a partir de imagens usando [TripoSG](https://github.com/VAST-AI-Research/TripoSG).
 
-    **Dicas:**
-    - Use imagens com objeto centralizado
-    - Fundo limpo ou use "Remover fundo"
-    - Mais passos = melhor qualidade (mais lento)
-    """)
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            input_image = gr.Image(
-                label="Imagem de Entrada",
-                type="pil",
-                height=400,
-            )
-
-            with gr.Accordion("Configurações", open=True):
-                remove_bg = gr.Checkbox(
-                    label="Remover fundo automaticamente",
-                    value=True,
-                )
-                num_steps = gr.Slider(
-                    label="Passos de inferência",
-                    minimum=20,
-                    maximum=100,
-                    value=50,
-                    step=5,
-                )
-                guidance_scale = gr.Slider(
-                    label="Guidance Scale",
-                    minimum=1.0,
-                    maximum=15.0,
-                    value=7.0,
-                    step=0.5,
-                )
-                seed = gr.Number(
-                    label="Seed (reprodutibilidade)",
-                    value=42,
-                    precision=0,
-                )
-                output_format = gr.Radio(
-                    label="Formato de saída",
-                    choices=["glb", "stl"],
-                    value="glb",
-                )
-
-            generate_btn = gr.Button("🚀 Gerar Modelo 3D", variant="primary", size="lg")
-
-        with gr.Column(scale=1):
-            # Usar File ao invés de Model3D para evitar bug de schema no HF Spaces
-            output_file = gr.File(label="Modelo 3D Gerado")
-            gr.Markdown("*Baixe o arquivo GLB/STL e visualize em um viewer 3D online como [glTF Viewer](https://gltf-viewer.donmccurdy.com/)*")
-
-    generate_btn.click(
-        fn=generate_3d,
-        inputs=[input_image, num_steps, guidance_scale, seed, remove_bg, output_format],
-        outputs=output_file,
-    )
-
-    gr.Markdown("""
-    ---
-    **Powered by:** [TripoSG](https://github.com/VAST-AI-Research/TripoSG) |
-    [RMBG-1.4](https://huggingface.co/briaai/RMBG-1.4)
-    """)
+    **Dicas:** Use imagens com objeto centralizado. Mais passos = melhor qualidade (mais lento).
+
+    Baixe o arquivo e visualize em [glTF Viewer](https://gltf-viewer.donmccurdy.com/)
+    """,
+    allow_flagging="never",
+)
 
 
 if __name__ == "__main__":

requirements.txt

@@ -23,7 +23,7 @@ scipy
 trimesh
 numpy>=1.22.3,<2.0.0
 
-# Interface - usar versão estável do Gradio (4.36.1 não tem bug do schema)
-gradio==4.36.1
+# Interface - usar versão antiga do Gradio sem bug de schema
+gradio==4.0.2
 
 # Nota: TripoSG será clonado via app.py (não é um pacote pip)