update

setup.sh

@@ -5,71 +5,19 @@ pip install torch==2.2.0+cu121 torchvision==0.17.0+cu121 torchaudio==2.2.0+cu121
 echo "Reinstalling NumPy with specific version..."
 pip install --force-reinstall -v "numpy==1.25.2"
 
-echo "Installing other requirements..."
-pip install -r requirements.txt
-
-# 创建延迟安装DISO的脚本
-cat > install_diso.py << 'EOF'
-import os
-import subprocess
-import torch
-import time
-
-# 等待GPU可用
-def wait_for_gpu():
-    max_attempts = 30
-    for i in range(max_attempts):
-        try:
-            # 尝试导入spaces模块以初始化GPU
-            import spaces
-            if torch.cuda.is_available():
-                print(f"GPU可用: {torch.cuda.get_device_name(0)}")
-                return True
-        except:
-            pass
-        print(f"等待GPU可用... ({i+1}/{max_attempts})")
-        time.sleep(10)
-    return False
 
-# 安装DISO包
-def install_diso():
-    print("开始安装DISO包...")
-    os.makedirs("diso_package", exist_ok=True)
-    os.chdir("diso_package")
-    subprocess.run(["git", "clone", "https://github.com/SarahWeiii/diso.git"])
-    os.chdir("diso")
-    subprocess.run(["pip", "install", "ninja"])
-    subprocess.run(["pip", "install", "-e", "."])
-    print("DISO安装完成")
+# Compile and install DISO package
+echo "Compiling and installing DISO..."
+pip install ninja
+mkdir -p diso_package
+cd diso_package
+git clone https://github.com/SarahWeiii/diso.git
+cd diso
+pip install -e .
+cd ..
 
-if __name__ == "__main__":
-    if wait_for_gpu():
-        install_diso()
-    else:
-        print("GPU未能在指定时间内可用，无法安装DISO")
-EOF
 
-echo "设置延迟安装DISO..."
-# 添加到app.py的启动脚本中运行安装脚本
-mv app.py app_original.py
-cat > app.py << 'EOF'
-import os
-import subprocess
-import sys
+echo "Setup completed successfully!"
 
-# 首先运行DISO安装脚本
-diso_installed_marker = ".diso_installed"
-if not os.path.exists(diso_installed_marker):
-    print("正在安装DISO包...")
-    try:
-        subprocess.run([sys.executable, "install_diso.py"])
-        with open(diso_installed_marker, "w") as f:
-            f.write("DISO installed")
-    except Exception as e:
-        print(f"DISO安装失败: {e}")
-
-# 然后执行原始app.py的内容
-exec(open("app_original.py").read())
-EOF
-
-echo "Setup completed successfully!"
+echo "Installing other requirements..."
+pip install -r requirements.txt