Update app.py
Browse files
app.py
CHANGED
@@ -60,12 +60,12 @@ 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=['
|
64 |
-
mod.bind(for_training=False, data_shapes=[('
|
65 |
mod.set_params(arg_params, aux_params, allow_missing=True, allow_extra=True)
|
66 |
|
67 |
-
title="
|
68 |
-
description="
|
69 |
|
70 |
examples=[['catonnx.jpg']]
|
71 |
gr.Interface(predict,gr.inputs.Image(type='filepath'),"label",title=title,description=description,examples=examples).launch(enable_queue=True)
|
|
|
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"
|
68 |
+
description="GoogLeNet is the name of a convolutional neural network for classification, which competed in the ImageNet Large Scale Visual Recognition Challenge in 2014."
|
69 |
|
70 |
examples=[['catonnx.jpg']]
|
71 |
gr.Interface(predict,gr.inputs.Image(type='filepath'),"label",title=title,description=description,examples=examples).launch(enable_queue=True)
|