Update app.py
Browse files
app.py
CHANGED
@@ -60,9 +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, label_names=None)
|
64 |
-
mod.bind(for_training=False, data_shapes=[('
|
65 |
-
label_shapes=mod._label_shapes)
|
66 |
mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)
|
67 |
|
68 |
title="MobileNet"
|
|
|
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="MobileNet"
|