Update app.py

app.py

@@ -60,9 +60,8 @@ if len(mx.test_utils.list_gpus())==0:
 else:
     ctx = mx.gpu(0)
 # Load module
-mod = mx.mod.Module(symbol=sym, context=ctx,data_names=['data_scale'], label_names=None)
-mod.bind(for_training=False, data_shapes=[('data_scale', (1,3,224,224))], 
-         label_shapes=mod._label_shapes)
+mod = mx.mod.Module(symbol=sym, context=ctx, data_names=['data_0'], label_names=None)
+mod.bind(for_training=False, data_shapes=[('data_0', (1,3,224,224))],label_shapes=mod._label_shapes)
 mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)
 
 title="ResNet"