Upload 2 files

app.py

@@ -9,15 +9,15 @@ os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")  # force CPU
 os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")  # avoid KNN/eval paths if supported
 os.environ.setdefault("GRADIO_TEMP_DIR", "/tmp/gradio")  # centralize gradio cache
 
-# Monkey patch to avoid CUDA errors
-try:
-    import nunchaku.utils
+# Monkey patch get_precision to avoid CUDA errors
+import nunchaku.utils
 
-    def safe_get_precision(*args, **kwargs):
-        return "fp32"  # force CPU precision
+def safe_get_precision(*args, **kwargs):
+    return "fp32"
 
-    nunchaku.utils.get_precision = safe_get_precision
+nunchaku.utils.get_precision = safe_get_precision
 
+try:
     import nunchaku.models.transformers.transformer_flux as tf
     tf.get_precision = safe_get_precision
 except Exception:
@@ -26,18 +26,11 @@ except Exception:
 import gradio as gr
 from worldgen import WorldGen
 
-# Lazy init
 _wg = None
 def get_worldgen():
     global _wg
     if _wg is None:
-        _wg = WorldGen(
-            mode="t2s",
-            device="cpu",
-            low_vram=True,
-            lora_path=None,
-            repo_id="mit-han-lab/svdq-int4-flux.1-dev"
-        )
+        _wg = WorldGen(mode="t2s", device="cpu", low_vram=True, lora_path=None)
     return _wg
 
 def _zip_directory(dir_path: Path) -> str:
@@ -92,9 +85,9 @@ 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:
+with gr.Blocks(title="WorldGen • Text -> 3D Scene") as demo:
     gr.Markdown(
-        """# WorldGen - Text → 3D
+        """# WorldGen - Text -> 3D
 - CPU-only fallback is enabled. For real runs, switch the Space to a GPU (T4/A10G).
 - Outputs are zipped for download; temp files are cleaned shortly after each run.
 """