Label Changes

app.py

@@ -97,8 +97,9 @@ def classify_info(image_1, text_1, image_2, text_2):
     ds = ds.batch(batch_size).prefetch(auto)
     output = model.predict(ds)
 
-    label = np.argmax(output)
-    return labels[label]
+    outputs = {labels[i], output[i] for i in range(len(output))}
+    #label = np.argmax(output)
+    return outputs #labels[label]
 
 
 model = from_pretrained_keras("keras-io/multimodal-entailment")
@@ -128,4 +129,4 @@ iface = gr.Interface(classify_info,
 	description = "Model for classifying whether image and text from one scenario complements the image and text from another scenario. They can be contradictory, implied or no entailment. Example images and text from the dataset in raw form !",
         article = "Author: <a href=\"https://huggingface.co/reichenbach\">Rishav Chandra Varma</a>")
 
-iface.launch(debug=True)
+iface.launch()