Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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")
|