Update app.py

app.py

@@ -1,38 +1,33 @@
+import os
 import gradio as gr
-import torch
-from shap_e.diffusion.sample import sample_latents
-from shap_e.diffusion.gaussian_diffusion import create_gaussian_diffusion
-from shap_e.models.text_to_3d import create_model_and_diffusion
-from shap_e.util.notebooks import decode_latent_mesh
+from huggingface_hub import InferenceClient
+import tempfile
+import trimesh
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+HF_TOKEN = os.getenv("HF_TOKEN")  # token environment
 
-print("Loading model...")
-model, diffusion = create_model_and_diffusion(batch_size=1, device=device)
-model.eval()
+client = InferenceClient(token=HF_TOKEN)
 
 def generate_3d(prompt):
-    with torch.no_grad():
-        latents = sample_latents(
-            batch_size=1,
-            model=model,
-            diffusion=diffusion,
-            guidance_scale=15.0,
-            model_kwargs=dict(texts=[prompt]),
-            progress=True,
-            device=device,
-        )
-        mesh = decode_latent_mesh(latents[0].cpu())
-        output_path = "/tmp/shape.glb"
-        with open(output_path, "wb") as f:
-            mesh.export(output_path)
-        return output_path
+    result = client.text_to_mesh(
+        model="Webaverse/Stable-Dreamfusion",
+        inputs={"prompt": prompt, "num_steps": 100},
+    )
+    mesh_bytes = result["output"]  # raw mesh bytes
+    temp = tempfile.NamedTemporaryFile(delete=False, suffix=".ply")
+    temp.write(mesh_bytes)
+    temp.flush()
+    # optionally convert to .glb
+    mesh = trimesh.load(temp.name)
+    glb_path = temp.name.replace(".ply", ".glb")
+    mesh.export(glb_path)
+    return glb_path
 
-gr.Interface(
+iface = gr.Interface(
     fn=generate_3d,
-    inputs=gr.Textbox(label="Prompt (contoh: 'a sci-fi turret tower')"),
-    outputs=gr.Model3D(label="3D Model Output (.glb)"),
-    title="Text to 3D - SHAP-E Generator",
-    description="Masukkan prompt dan hasilkan file 3D untuk digunakan di game engine seperti Unity/Unreal"
-).launch()
-
+    inputs=gr.Textbox(label="Prompt Game Asset"),
+    outputs=gr.Model3D(label="Generated 3D Asset (.glb)"),
+    title="Text‑to‑3D Game Asset (Dreamfusion)",
+    description="Masukkan prompt, hasil akan berupa file .glb yang siap impor ke game engine."
+)
+iface.launch()