yansong1616 commited on
Commit
fbb1647
·
verified ·
1 Parent(s): 1dd77e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ plt.ion()
30
 
31
 
32
  # 添加 sam2 模块路径
 
33
  sys.path.append(os.path.join(os.path.dirname(__file__), 'SAM2'))
34
 
35
  torch.backends.cuda.matmul.allow_tf32 = True # for gpu >= Ampere and pytorch >= 1.12
 
30
 
31
 
32
  # 添加 sam2 模块路径
33
+ import sys
34
  sys.path.append(os.path.join(os.path.dirname(__file__), 'SAM2'))
35
 
36
  torch.backends.cuda.matmul.allow_tf32 = True # for gpu >= Ampere and pytorch >= 1.12