Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.10-slim as builder
+FROM python:3.10-slim
 
 # System dependencies
 RUN apt-get update && apt-get install -y \
@@ -15,16 +15,8 @@ RUN pip install --no-cache-dir torch==2.2.0 --index-url https://download.pytorch
 RUN pip install --no-cache-dir \
     transformers==4.40.0 \
     gradio==4.24.0 \
-    accelerate==0.29.0
-
-# Runtime image
-FROM python:3.10-slim
-WORKDIR /app
-
-# Copy only necessary files
-COPY --from=builder /usr/local/lib/python3.10/site-packages /usr/local/lib/python3.10/site-packages
-COPY --from=builder /usr/local/bin/git-lfs /usr/local/bin/
-COPY --from=builder /app/app.py .
+    accelerate==0.29.0 \
+    fastapi
 
 # Environment variables
 ENV PYTHONUNBUFFERED=1 \
@@ -33,9 +25,5 @@ ENV PYTHONUNBUFFERED=1 \
     HF_TOKEN="" \
     DEVICE="cpu"
 
-# Health check
-HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:7860 || exit 1
-
 EXPOSE 7860
 CMD ["python", "app.py"]