akhaliq HF staff commited on
Commit
7aaf95a
1 Parent(s): 8372848

Update app.py

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