new

app.py

@@ -5,7 +5,7 @@ import spaces
 import ctypes
 import shlex
 import torch
-subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
+
 subprocess.run(
     shlex.split(
         "pip install ./custom_diffusers --force-reinstall --no-deps"
@@ -93,7 +93,6 @@ def check_gpu():
     os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
     # os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
     os.environ['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
-    subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
     # 显式加载 libnvrtc.so.12
     cuda_lib_path = "/usr/local/cuda-12.1/lib64/libnvrtc.so.12"
     try:
@@ -285,7 +284,7 @@ def image_to_base64(image_path):
     """Converts an image file to a base64-encoded string."""
     with open(image_path, "rb") as img_file:
         return base64.b64encode(img_file.read()).decode('utf-8')
-
+subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 def main():
 
     torch.set_grad_enabled(False)