onuralpszr commited on
Commit
9064b3b
Β·
verified Β·
1 Parent(s): 66c6695

fix: 🐞 return runner itself

Browse files

Signed-off-by: Onuralp SEZER <[email protected]>

Files changed (1) hide show
  1. app.py +1 -0
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,