Upload 2 files

Dockerfile

@@ -3,6 +3,8 @@ FROM python:3.9-slim
 
 # Set the working directory
 WORKDIR /app
+RUN mkdir -p /tmp/Ultralytics && chmod -R 777 /tmp/Ultralytics
+ENV YOLO_CONFIG_DIR=/tmp/Ultralytics
 
 # Install system dependencies
 # We use libgl1 and libglib2.0-0 which are the modern equivalents

requirements.txt

@@ -5,4 +5,5 @@ python-multipart
 pillow
 numpy
 torch
-torchvision
+torchvision
+transformers