akhaliq HF staff commited on
Commit
871866d
1 Parent(s): 57b24df

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=['conv1/7x7_s2_w_0'], label_names=None)
64
- mod.bind(for_training=False, data_shapes=[('conv1/7x7_s2_w_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="GoogleNet"
 
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="GoogleNet"