Update app.py

app.py

@@ -16,9 +16,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 gpu_checkpoint_path = "efficientsam_s_gpu.jit"
 cpu_checkpoint_path = "efficientsam_s_cpu.jit"
+ti_gpu_checkpoint_path = "efficientsam_ti_gpu.jit"
 
 if torch.cuda.is_available():
-    model = torch.jit.load(gpu_checkpoint_path)
+    model = torch.jit.load(ti_gpu_checkpoint_path)
+    # model = torch.jit.load(gpu_checkpoint_path)
 else:
     model = torch.jit.load(cpu_checkpoint_path)
 model.eval()