akhaliq HF staff commited on
Commit
fd009be
1 Parent(s): 8dc7edb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ if len(mx.test_utils.list_gpus())==0:
60
  else:
61
  ctx = mx.gpu(0)
62
  # Load module
63
- mod = mx.mod.Module(symbol=sym, context=ctx,data_names=['data_scale']\, label_names=None)
64
  mod.bind(for_training=False, data_shapes=[('data_scale', (1,3,224,224))],
65
  label_shapes=mod._label_shapes)
66
  mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)
 
60
  else:
61
  ctx = mx.gpu(0)
62
  # Load module
63
+ mod = mx.mod.Module(symbol=sym, context=ctx,data_names=['data_scale'], label_names=None)
64
  mod.bind(for_training=False, data_shapes=[('data_scale', (1,3,224,224))],
65
  label_shapes=mod._label_shapes)
66
  mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)