Upgrade to PyTorch 2.2.0 for ultralytics compatibility

Dockerfile

@@ -2,7 +2,7 @@
 # FarmEyes - HuggingFace Spaces Dockerfile (Transformers Version)
 # =============================================================================
 
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+FROM pytorch/pytorch:2.2.0-cuda11.8-cudnn8-runtime
 
 ENV PYTHONUNBUFFERED=1
 ENV HOST=0.0.0.0
@@ -25,9 +25,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Upgrade pip
 RUN pip install --upgrade pip setuptools wheel
 
-# Install ultralytics with verbose output
-RUN pip install ultralytics==8.0.200 && \
-    python -c "import ultralytics; print('Ultralytics version:', ultralytics.__version__)"
+# Install ultralytics
+RUN pip install ultralytics
 
 # Copy requirements and install
 COPY requirements.txt .
@@ -39,9 +38,6 @@ COPY . .
 # Create directories
 RUN mkdir -p /app/uploads /app/temp
 
-# Final check - list installed packages
-RUN pip list | grep -i ultra
-
 EXPOSE 7860
 
 CMD ["python", "main.py"]