Update app.py

app.py

@@ -54,11 +54,13 @@ def predict(image):
     for i in range(len(model_all.names)):
         count = class_counts_all[i]
         avg_conf = np.mean(class_confidences_all[i]) if class_confidences_all[i] else 0
-        output += f"{model_all.names[i]}: {count} detections (Avg. Confidence: {avg_conf:.2f})\n"
+        if count > 0:  # Only print classes with detections
+            output += f"{model_all.names[i]}: {count} detections (Avg. Confidence: {avg_conf:.2f})\n"
 
     # Add number plate detection results from the second model
-    avg_np_conf = np.mean(np_confidences) if np_confidences else 0
-    output += f"Number Plates : {np_count} detections (Avg. Confidence: {avg_np_conf:.2f})\n"
+    if np_count > 0:  # Only print number plate detection if there are any
+        avg_np_conf = np.mean(np_confidences) if np_confidences else 0
+        output += f"Number Plates: {np_count} detections (Avg. Confidence: {avg_np_conf:.2f})\n"
 
     return im, output
 
@@ -75,4 +77,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch(share='True')
+iface.launch(share=True)