Update app.py

app.py

@@ -305,6 +305,14 @@ def reconstruct_3d_model(images, prompt):
 # Gradio 接口函数
 @spaces.GPU
 def gradio_pipeline(prompt, seed):
+    import ctypes
+    # 显式加载 libnvrtc.so.12
+    cuda_lib_path = "/usr/local/cuda-12.1/lib64/libnvrtc.so.12"
+    try:
+        ctypes.CDLL(cuda_lib_path, mode=ctypes.RTLD_GLOBAL)
+        print(f"Successfully preloaded {cuda_lib_path}")
+    except OSError as e:
+        print(f"Failed to preload {cuda_lib_path}: {e}")
     # 生成多视图图像
     # rgb_normal_grid = generate_multi_view_images(prompt, seed)
     rgb_normal_grid = np.load("rgb_normal_grid.npy")