joheras commited on
Commit
b9146e0
1 Parent(s): fc7adb5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def classify_image(image):
21
  prediction = softmax(pred)[0]
22
  return {labels[i]: float(prediction[i]) for i in range(100)}
23
 
24
- image = gr.inputs.Image(shape(32,32))
25
  label = gr.outputs.Label(num_top_classes=5)
26
 
27
  iface = gr.Interface(classify_image,image,label,
 
21
  prediction = softmax(pred)[0]
22
  return {labels[i]: float(prediction[i]) for i in range(100)}
23
 
24
+ image = gr.inputs.Image(shape=(32,32))
25
  label = gr.outputs.Label(num_top_classes=5)
26
 
27
  iface = gr.Interface(classify_image,image,label,