multi

app.py

@@ -57,7 +57,7 @@ def classify_image_multi(inp):
   inp = inp.reshape((-1, 224, 224, 3))
   inp = tf.keras.applications.densenet.preprocess_input(inp)
   prediction = model_multi.predict(inp)
-  confidences = {labels_multi[i]: float(prediction[0][i]) for i in range(3)}
+  confidences = {labels_multi[i]: float(prediction[0][i]) for i in range(len(prediction[0]))}
   return confidences
     
 binary_interface = gr.Interface(fn=classify_image_binary,