Updates the checkpoint path
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import gradio as gr
|
|
| 3 |
from src.model import DRModel
|
| 4 |
from torchvision import transforms as T
|
| 5 |
|
| 6 |
-
CHECKPOINT_PATH = "
|
| 7 |
model = DRModel.load_from_checkpoint(CHECKPOINT_PATH)
|
| 8 |
|
| 9 |
labels = {
|
|
|
|
| 3 |
from src.model import DRModel
|
| 4 |
from torchvision import transforms as T
|
| 5 |
|
| 6 |
+
CHECKPOINT_PATH = "artifacts/dr-model.ckpt"
|
| 7 |
model = DRModel.load_from_checkpoint(CHECKPOINT_PATH)
|
| 8 |
|
| 9 |
labels = {
|