new

app.py

@@ -153,6 +153,7 @@ def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_
     init_image = None
     # if init_image_path is not None:
     #     init_image = Image.open(init_image_path)
+    subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
     result = k3d_wrapper.generate_3d_bundle_image_text( 
                                       prompt,
                                       image=init_image,