Label Changes

app.py

@@ -97,7 +97,11 @@ def classify_info(image_1, text_1, image_2, text_2):
     ds = ds.batch(batch_size).prefetch(auto)
     output = model.predict(ds)
 
-    outputs = {(labels[i], output[i]) for i in range(len(output))}
+    outputs = dict()
+    
+    outputs[labels[0]] = float(output[0][0])
+    outputs[labels[1]] = float(output[0][1])
+    outputs[labels[2]] = float(output[0][2])
     #label = np.argmax(output)
     return outputs #labels[label]