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,
|