akhaliq HF staff commited on
Commit
3d4a5e7
1 Parent(s): eae3af0

Update app.py

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