Update app.py

app.py

@@ -106,8 +106,8 @@ def predict(image):
         "front": front_result,
         "back": back_result
     }
-
-    return cropped_images, labels, api_results, detected_image
+    single_image = cropped_images.get("front") or cropped_images.get("back") or detected_image   
+    return single_image, labels, api_results
 
 # ---------------- Gradio Interface ---------------- #
 iface = gr.Interface(