Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ else:
|
|
61 |
ctx = mx.gpu(0)
|
62 |
# Load module
|
63 |
|
64 |
-
mod = mx.mod.Module(symbol=sym, context=ctx,data_names=['
|
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)
|
|
|
61 |
ctx = mx.gpu(0)
|
62 |
# Load module
|
63 |
|
64 |
+
mod = mx.mod.Module(symbol=sym, context=ctx,data_names=['data_0'],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)
|