Update app.py

app.py

@@ -20,10 +20,7 @@ def count_packages(image, model_choice):
     # Get the number of detected boxes (packages)
     package_count = len(results[0].boxes)
     
-    # Annotate the image with the detected boxes
-    annotated_image = results[0].plot()  # Plot the detections on the image
     
-    return package_count, annotated_image
 
 # Gradio interface with model selection dropdown
 inputs = [