Update Dockerfile

Dockerfile

@@ -1,38 +1,13 @@
-# Use Python 3.9 as base image
 FROM python:3.9
 
-# Create a non-root user
 RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
-# Set environment variables for model caching
-ENV TRANSFORMERS_CACHE="/home/user/.cache/huggingface"
-ENV TORCH_HOME="/home/user/.cache/torch"
-
-# Create cache directories
-RUN mkdir -p $TRANSFORMERS_CACHE $TORCH_HOME && chmod -R 777 $TRANSFORMERS_CACHE $TORCH_HOME
-
-# Preload and cache models
-RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; \
-               AutoModelForCausalLM.from_pretrained('microsoft/phi-2', cache_dir='$TRANSFORMERS_CACHE'); \
-               AutoTokenizer.from_pretrained('microsoft/phi-2', cache_dir='$TRANSFORMERS_CACHE')" \
-    && python -c "from torchvision.models.detection import fasterrcnn_resnet50_fpn, FasterRCNN_ResNet50_FPN_Weights; \
-                  weights = FasterRCNN_ResNet50_FPN_Weights.DEFAULT; \
-                  model = fasterrcnn_resnet50_fpn(weights=weights, cache_dir='$TORCH_HOME')"
-
-# Set working directory
 WORKDIR /app
 
-# Copy requirements and install dependencies
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Copy application code
 COPY --chown=user . /app
-
-# Expose the port
-EXPOSE 7860
-
-# Start the FastAPI app with Uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]