Update model_yolov5.py

model_yolov5.py

@@ -12,11 +12,11 @@ def load_model(model_path, img_size=640):
     return model
 
 
-# @spaces.GPU
+@spaces.GPU
 def inference(model, image):
     """Run inference on image and return annotated image."""
     results = model(image, size=model.img_size)
-    print(results)
+    print(f"{results=}")
     annotator = Annotator(np.asarray(image))
     for *box, _, cls in reversed(results.pred[0]):
         # label = f'{model.names[int(cls)]} {conf:.2f}'