Spaces:
Runtime error
Runtime error
Nick
commited on
Commit
·
5b36e55
1
Parent(s):
d8b5000
Move Gradio app.py to root for Hugging Face Spaces compatibility; original removed from gradio_app/
Browse files
gradio_app/app.py → app.py
RENAMED
|
@@ -14,10 +14,10 @@ from models.unet_depthwise_nano import UNetDepthwiseNano
|
|
| 14 |
from utils.checkpoint import load_checkpoint
|
| 15 |
|
| 16 |
MODEL_PATHS = {
|
| 17 |
-
"unet": "
|
| 18 |
-
"unet_depthwise": "
|
| 19 |
-
"unet_depthwise_small": "
|
| 20 |
-
"unet_depthwise_nano": "
|
| 21 |
}
|
| 22 |
|
| 23 |
IMG_SIZE = 256
|
|
|
|
| 14 |
from utils.checkpoint import load_checkpoint
|
| 15 |
|
| 16 |
MODEL_PATHS = {
|
| 17 |
+
"unet": "checkpoints/unet_best.pt",
|
| 18 |
+
"unet_depthwise": "checkpoints/unet_depthwise_best.pt",
|
| 19 |
+
"unet_depthwise_small": "checkpoints/unet_depthwise_small_best.pt",
|
| 20 |
+
"unet_depthwise_nano": "checkpoints/unet_depthwise_nano_best.pt"
|
| 21 |
}
|
| 22 |
|
| 23 |
IMG_SIZE = 256
|