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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -60,8 +60,9 @@ 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, label_names=None)
64
- mod.bind(for_training=False, data_shapes=[('data', (1,3,224,224))],
 
65
  label_shapes=mod._label_shapes)
66
  mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)
67
 
 
60
  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)
68