Fix config error and optimize Docker setup with ConvNextV2 model

.spacesconfig.yaml

@@ -0,0 +1,2 @@
+sdk: docker
+app_port: 7860

Dockerfile

@@ -1,11 +1,22 @@
-FROM python:3.9
+FROM python:3.9-slim
 
 WORKDIR /code
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install Python dependencies
 COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Copy application code
+COPY ./app.py /code/app.py
 
-COPY ./app.py /code/
+# Expose port
+EXPOSE 7860
 
+# Run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,8 +1,7 @@
-fastapi
-uvicorn[standard]
-python-multipart
-pillow
-torch
-torchvision
-transformers
-numpy
+fastapi==0.104.1
+uvicorn[standard]==0.24.0
+python-multipart==0.0.6
+pillow==10.1.0
+torch==2.1.0
+transformers==4.35.0
+numpy==1.24.3