Spaces:
Runtime error
Runtime error
fix: π return runner itself
Browse filesSigned-off-by: Onuralp SEZER <[email protected]>
app.py
CHANGED
|
@@ -51,6 +51,7 @@ def load_runner():
|
|
| 51 |
pipeline = cfg.test_dataloader.dataset.pipeline
|
| 52 |
runner.pipeline = Compose(pipeline)
|
| 53 |
runner.model.eval()
|
|
|
|
| 54 |
|
| 55 |
def run_image(
|
| 56 |
input_image,
|
|
|
|
| 51 |
pipeline = cfg.test_dataloader.dataset.pipeline
|
| 52 |
runner.pipeline = Compose(pipeline)
|
| 53 |
runner.model.eval()
|
| 54 |
+
return runner
|
| 55 |
|
| 56 |
def run_image(
|
| 57 |
input_image,
|