johnlockejrr commited on
Commit
e59fb5f
·
verified ·
1 Parent(s): 0552a9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def predict(model_name, input_img):
61
  experiment_dirname="",
62
  )
63
 
64
- data_args = DataArgs(batch_size=batch_size, color_mode="L", num_workers=NUM_WORKERS)
65
  trainer_args = TrainerArgs(progress_bar_refresh_rate=0)
66
  decode_args = DecodeArgs(
67
  include_img_ids=True,
 
61
  experiment_dirname="",
62
  )
63
 
64
+ data_args = DataArgs(batch_size=batch_size, color_mode="L")
65
  trainer_args = TrainerArgs(progress_bar_refresh_rate=0)
66
  decode_args = DecodeArgs(
67
  include_img_ids=True,