YinuoGuo27 commited on
Commit
0da1d9d
·
verified ·
1 Parent(s): 64dbe94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -9,6 +9,9 @@ from huggingface_hub import snapshot_download
9
  import spaces
10
  import tensorrt
11
  print(tensorrt.__path__)
 
 
 
12
 
13
 
14
  repo_id = "ChaolongYang/KDTalker"
 
9
  import spaces
10
  import tensorrt
11
  print(tensorrt.__path__)
12
+ plugin_path = os.path.join(trt_path, "libgrid_sample_3d_plugin.so")
13
+ if os.path.exists(plugin_path):
14
+ print("✅ libgrid_sample_3d_plugin.so is found at:", plugin_path)
15
 
16
 
17
  repo_id = "ChaolongYang/KDTalker"