Update Dockerfile

Dockerfile

@@ -33,9 +33,13 @@ RUN mkdir -p /app/{weights,imgs} && \
     chmod -R 777 /app
 
 # Application code
-COPY main.py utils.py download_models.py ./
-COPY util ./util 
-# Download models
+COPY main.py utils.py download_models.py ./  # Copy your app code
+COPY util ./util                          # Copy the util folder
+
+# Ensure weights folder is also copied
+COPY weights /app/weights                  # Ensure the weights folder with models is copied
+
+# Download models if needed
 RUN python3 download_models.py
 
 # Set up user
@@ -43,4 +47,5 @@ RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Set the entry point to run the app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]