Update app.py

app.py

@@ -6,6 +6,10 @@ from PIL import Image
 from roboflow import Roboflow
 from ultralytics import YOLO 
 
+def summary(model, input_size):
+    from torchsummary import summary
+    summary(model, input_size=input_size)
+
 def load_model(file_path):
     # Load the Roboflow model
     rf = Roboflow(api_key="K1TXQnJq7EE7yoCf1g3C")