dongdongzi commited on
Commit
8177c52
·
verified ·
1 Parent(s): 46e7995

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,8 +21,10 @@ ti_gpu_checkpoint_path = "efficientsam_ti_gpu.jit"
21
  if torch.cuda.is_available():
22
  model = torch.jit.load(ti_gpu_checkpoint_path)
23
  # model = torch.jit.load(gpu_checkpoint_path)
 
24
  else:
25
  model = torch.jit.load(cpu_checkpoint_path)
 
26
  model.eval()
27
 
28
  # Description
 
21
  if torch.cuda.is_available():
22
  model = torch.jit.load(ti_gpu_checkpoint_path)
23
  # model = torch.jit.load(gpu_checkpoint_path)
24
+ print(f"using model {ti_gpu_checkpoint_path}")
25
  else:
26
  model = torch.jit.load(cpu_checkpoint_path)
27
+ print(f"using model {cpu_checkpoint_path}")
28
  model.eval()
29
 
30
  # Description