repo

app.py

@@ -8,8 +8,7 @@ from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image
 import gradio as gr
 from huggingface_hub import hf_hub_download
-import yolov9
-from yolov9.models.yolo import Model
+from yolo import Model
 
 model = Model('best.pt')
 

{yolov9/models → models}/common.py

@@ -22,7 +22,7 @@ from IPython.display import display
 from PIL import Image
 from torch.cuda import amp
 
-# from utils import TryExcept
+from utils import TryExcept
 from utils.dataloaders import exif_transpose, letterbox
 from utils.general import (LOGGER, ROOT, Profile, check_requirements, check_suffix, check_version, colorstr,
                            increment_path, is_notebook, make_divisible, non_max_suppression, scale_boxes,
@@ -1123,7 +1123,7 @@ class Detections:
                 LOGGER.info(f'Saved results to {save_dir}\n')
             return crops
 
-    # @TryExcept('Showing images is not supported in this environment')
+    @TryExcept('Showing images is not supported in this environment')
     def show(self, labels=True):
         self._run(show=True, labels=labels)  # show results