Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def teed_process_image(image):
|
|
| 58 |
args.input_val_dir = './teed_tmp'
|
| 59 |
args.output_dir = './output/teed_imgs'
|
| 60 |
|
| 61 |
-
checkpoint_path = './
|
| 62 |
if not os.path.exists(checkpoint_path):
|
| 63 |
raise FileNotFoundError(f"Checkpoint file not found: {checkpoint_path}")
|
| 64 |
|
|
|
|
| 58 |
args.input_val_dir = './teed_tmp'
|
| 59 |
args.output_dir = './output/teed_imgs'
|
| 60 |
|
| 61 |
+
checkpoint_path = './TEED/checkpoints/BIPED/5/5_model.pth'
|
| 62 |
if not os.path.exists(checkpoint_path):
|
| 63 |
raise FileNotFoundError(f"Checkpoint file not found: {checkpoint_path}")
|
| 64 |
|