Update app.py

app.py

@@ -20,7 +20,7 @@ def classify_image(image):
   resized_image = tf.image.resize(
     tf.convert_to_tensor([image]), size=(IMAGE_SIZE, IMAGE_SIZE))
   pred = model.predict(resized_image)
-  prediction = softmax(pred)
+  prediction = softmax(pred)[0]
   return {labels[i]: float(prediction[i]) for i in range(100)} 
   
 image = gr.inputs.Image()