Update app.py

app.py

@@ -3,7 +3,6 @@ import matplotlib.pyplot as plt
 import numpy as np
 from collections import namedtuple
 from mxnet.gluon.data.vision import transforms
-from mxnet.contrib.onnx.onnx2mx.import_model import import_model
 import os
 import gradio as gr
 
@@ -56,7 +55,7 @@ def predict(path):
     a = np.argsort(-outputs[0].flatten())
     results = {}
     for i in a[0:5]:
-        results[label[i]]=outputs[0][0][i]
+        results[labels[i]]=float(outputs[0][0][i])
     return results
        
 
@@ -64,4 +63,4 @@ title="AlexNet"
 description="AlexNet is the name of a convolutional neural network for classification, which competed in the ImageNet Large Scale Visual Recognition Challenge in 2012."
 
 examples=[['catonnx.jpg']]
-gr.Interface(predict,gr.inputs.Image(type='filepath'),"label",title=title,description=description,examples=examples).launch(enable_queue=True)
+gr.Interface(predict,gr.inputs.Image(type='filepath'),"label",title=title,description=description,examples=examples).launch(enable_queue=True,debug=True)