Update main.py

main.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from ultralytics import YOLO
-from ultralytics.solutions import ai_gym
+import ai_gym
 import cv2
 import tempfile
 import os