Update app.py

app.py

@@ -1,13 +1,13 @@
 import cv2
 import torch
-import gradio as gr  # Ensure correct import with alias
+import gradio as gr
 import numpy as np
-from ultralytics import YOLO
+from ultralytics import YOLO, __version__ as ultralytics_version  # Import ultralytics version
 
 # Debug: Check environment
 print(f"Torch version: {torch.__version__}")
 print(f"Gradio version: {gr.__version__}")
-print(f"Ultralytics version: {YOLO.__version__}")
+print(f"Ultralytics version: {ultralytics_version}")  # Corrected version access
 print(f"CUDA available: {torch.cuda.is_available()}")
 
 # Load YOLOv8 model