microhan commited on
Commit
43efe1e
1 Parent(s): 565cb88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ from utils.tools_gradio import fast_process
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
- gpu_checkpoint_path = "efficientsam_s_gpu.jit"
18
- cpu_checkpoint_path = "efficientsam_s_cpu.jit"
19
 
20
  if torch.cuda.is_available():
21
  model = torch.jit.load(gpu_checkpoint_path)
 
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
+ gpu_checkpoint_path = "efficientsam_ti_gpu.jit"
18
+ cpu_checkpoint_path = "efficientsam_ti_cpu.jit"
19
 
20
  if torch.cuda.is_available():
21
  model = torch.jit.load(gpu_checkpoint_path)