Update Dockerfile

Dockerfile

@@ -1,18 +1,24 @@
-FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
+FROM python:3.10-slim
 
-RUN apt-get update && apt-get install -y \
-    python3 python3-pip git && \
-    ln -s /usr/bin/python3 /usr/bin/python
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
-COPY requirements.txt .
+# Install dependencies
+RUN apt-get update && apt-get install -y git
+
+# Install Python packages
 RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
+RUN pip install fastapi uvicorn transformers torch
 
+# Set workdir
 WORKDIR /app
-ENV TRANSFORMERS_CACHE=/app/cache
-RUN mkdir -p /app/cache
 
-COPY app.py .
+# Copy your app code
+COPY . .
 
+# Expose port
 EXPOSE 7860
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Run app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]