refactored the code

This commit is contained in:
Bala-Vignesh-Reddy 2025-03-22 23:27:13 +05:30
parent 6695bac0cd
commit 8b9d980dff

View File

@ -21,6 +21,7 @@ import torch
import torch.nn as nn
from PIL import Image
CHECK_PYTORCH_18 = torch.__version__.startswith("1.8")
# Import 'ultralytics' package or install if missing
try:
import ultralytics