Update app.py

app.py

@@ -97,7 +97,7 @@ 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 = {(labels[i], output[i]) for i in range(len(output))}
     #label = np.argmax(output)
     return outputs #labels[label]