Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import shutil, tempfile
|
| 2 |
from pathlib import Path
|
| 3 |
import gradio as gr, torch
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "" # 強制 CPU
|
| 3 |
+
os.environ["WORLDGEN_DISABLE_EVAL"] = "1" # 若程式支援,避免 KNN
|
| 4 |
+
|
| 5 |
+
import gradio as gr
|
| 6 |
+
from worldgen import WorldGen
|
| 7 |
+
|
| 8 |
+
wg = WorldGen(mode="t2s", device="cpu", low_vram=True)
|
| 9 |
+
|
| 10 |
+
gr.Markdown("""# WorldGen — 文字 -> 3D 場景- 僅啟用 Text -> 3D。- Free CPU 會很慢,建議升級 GPU(T4/A10G)。""")
|
| 11 |
+
# 其餘邏輯同前;記得回傳後用延時 Thread 清理臨時資料夾,避免 50GB 被用滿
|
| 12 |
+
|
| 13 |
import shutil, tempfile
|
| 14 |
from pathlib import Path
|
| 15 |
import gradio as gr, torch
|