debug

app.py

@@ -187,6 +187,11 @@ def text_to_detailed(prompt, seed=None):
     subprocess.run(['nvidia-smi'])  # Test if CUDA is available
     print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
     print("Device count:", torch.cuda.device_count()) 
+    subprocess.run(
+        shlex.split(
+            "pip install ./extension/nvdiffrast-0.3.1+torch-py3-none-any.whl --force-reinstall --no-deps"
+        )
+    )
     import nvdiffrast.torch as dr
     dr.RasterizeCudaContext()
     print("nvdiffrast initialized successfully")