shimu0215 commited on
Commit
de64dcf
·
verified ·
1 Parent(s): 2d724c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ class Config:
28
  class ModelManager:
29
  @staticmethod
30
  def load_model(checkpoint_name: str):
31
- checkpoint_path = os.path.join(shimu0215/seg’, Config.CHECKPOINTS[checkpoint_name])
32
  model = torch.jit.load(checkpoint_path)
33
  model.eval()
34
  model.to("cuda")
 
28
  class ModelManager:
29
  @staticmethod
30
  def load_model(checkpoint_name: str):
31
+ checkpoint_path = os.path.join('shimu0215/seg', Config.CHECKPOINTS[checkpoint_name])
32
  model = torch.jit.load(checkpoint_path)
33
  model.eval()
34
  model.to("cuda")