Update app.py

app.py

@@ -79,6 +79,24 @@ def inference(image):
 
     result_image = get_image(top4[0])
     result = f"{result_image['label']} {result_image['timestamp']}"
+
+    n=2
+    plt.figure(figsize=(8, 8))
+    for i, (image1, image2) in enumerate(zip(top4[:2], top4[2:])):
+        ax = plt.subplot(2, n, i + 1)
+        image1 = get_image(image1)["image"]
+        image2 = get_image(image2)["image"]
+    
+        plt.imshow(image1)
+        plt.gray()
+        ax.get_xaxis().set_visible(False)
+        ax.get_yaxis().set_visible(False)
+    
+        ax = plt.subplot(2, n, i + 1 + n)
+        plt.imshow(image2)
+        plt.gray()
+        ax.get_xaxis().set_visible(False)
+        ax.get_yaxis().set_visible(False)
     
     return result