should work now

app.py

@@ -2,9 +2,7 @@ import os
 import sys
 import cv2
 import time
-import tyro
 import json
-import kiui
 import tqdm
 import torch
 import mcubes
@@ -22,6 +20,13 @@ from huggingface_hub import hf_hub_download
 
 sys.path.append("3DTopia")
 
+os.system("pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch")
+os.system("pip install git+https://github.com/NVlabs/nvdiffrast")
+os.system("pip install git+https://github.com/3DTopia/threefiner")
+
+import tyro
+import kiui
+
 from ldm.models.diffusion.ddim import DDIMSampler
 from ldm.models.diffusion.plms import PLMSSampler
 from ldm.models.diffusion.dpm_solver import DPMSolverSampler
@@ -110,6 +115,7 @@ opt.save = GRADIO_SAVE_PATH_MESH
 opt.prompt = ''
 opt.text_dir = True
 opt.front_dir = '+z'
+opt.force_cuda_rast = True
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 gui = GUI(opt)
 ###################################### INIT STAGE 2 #########################################
@@ -135,7 +141,7 @@ def add_text(rgb, caption):
 
 def marching_cube(b, text, global_info):
     # prepare volumn for marching cube
-    res = 128
+    res = 64
     assert 'decode_res' in global_info
     decode_res = global_info['decode_res']
     c_list = torch.linspace(-1.2, 1.2, steps=res)
@@ -358,6 +364,7 @@ def process_stage2(input_model, input_text, input_dir, iters, output_model, outp
 
 markdown=f'''
   # 3DTopia
+  ![](https://visitor-badge.laobi.icu/badge?page_id=3DTopia.3DTopia.gradio)
   A two-stage text-to-3D generation model. The first stage uses diffusion model to quickly generate candidates. The second stage refines the assets chosen from the first stage.
 
   ### Usage:

requirements.txt

@@ -1,6 +1,7 @@
-torch
-torchvision
-torchaudio
+torch==1.13.1+cu117
+torchvision==0.14.1+cu117
+torchaudio==0.13.1
+--extra-index-url https://download.pytorch.org/whl/cu117
 pytorch-lightning
 numpy
 tqdm
@@ -55,6 +56,5 @@ vit-pytorch
 wandb
 wcwidth
 zipp
-git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
-git+https://github.com/NVlabs/nvdiffrast
-git+https://github.com/3DTopia/threefiner
+kiui
+accelerate