Upload 2 files

app.py

@@ -5,18 +5,15 @@ import threading
 from pathlib import Path
 
 # === Runtime environment tweaks ===
-os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")  # force CPU
+os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")
 os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")
 os.environ.setdefault("GRADIO_TEMP_DIR", "/tmp/gradio")
 
 # Monkey patch to avoid CUDA errors
 import nunchaku.utils
-
 def safe_get_precision(*args, **kwargs):
     return "fp32"
-
 nunchaku.utils.get_precision = safe_get_precision
-
 try:
     import nunchaku.models.transformers.transformer_flux as tf
     tf.get_precision = safe_get_precision
@@ -31,7 +28,7 @@ _wg = None
 def get_worldgen():
     global _wg
     if _wg is None:
-        # Download model file from correct repo
+        # Confirm model file is downloadable
         model_path = hf_hub_download(
             repo_id="mit-han-lab/svdq-int4-flux.1-dev",
             filename="unquantized_layers.safetensors",
@@ -79,14 +76,17 @@ def generate_from_text(prompt: str):
     return zip_path, "Done. Download the ZIP (contains 3D assets)."
 
 with gr.Blocks(title="WorldGen • Text -> 3D Scene") as demo:
-    gr.Markdown("# WorldGen - Text -> 3D\n- CPU-only fallback is enabled.\n- Outputs are zipped for download.")
+    gr.Markdown("# WorldGen - Text -> 3D")
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Scene prompt (English)", placeholder="a neon-lit cyberpunk street", lines=3)
+            prompt = gr.Textbox(
+                label="Scene prompt (English)",
+                placeholder="a neon-lit cyberpunk street with rain reflections, highly detailed",
+                lines=3,
+            )
             btn = gr.Button("Generate 3D Scene")
         with gr.Column():
             out_zip = gr.File(label="Download output (ZIP)")
             status = gr.Markdown()
     btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
-
 demo.launch()