Update Dockerfile

Dockerfile

@@ -1,21 +1,22 @@
-# 1. Use a lightweight Python base
+# 1. Base Image
 FROM python:3.11-slim
 
 # 2. Set the working directory
 WORKDIR /app
 
-# 3. Install PyTorch (CPU version)
-RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
+# 3. Install dependencies (Pre-downloading the NEW model we discussed)
+RUN pip install --no-cache-dir fastapi uvicorn transformers pillow pydantic requests torch torchvision --index-url https://download.pytorch.org/whl/cpu
+RUN python -c "from transformers import pipeline; pipeline('image-classification', model='prithivMLmods/deepfake-detector-model-v1')"
 
-# 4. Install API dependencies
-RUN pip install --no-cache-dir fastapi uvicorn transformers pillow pydantic requests
+# 4. Copy your app.py specifically into the current WORKDIR (/app)
+# This ensures uvicorn can find the "app" module
+COPY app.py /app/app.py
 
-# 5. THE CHEAT CODE: Pre-download the Deepfake weights
-# Change this line in your Dockerfile
-# Change this line in your Dockerfile
-RUN python -c "from transformers import pipeline; pipeline('image-classification', model='prithivMLmods/deepfake-detector-model-v1')"
-# 7. Expose port
+# 5. Set environment variable to make sure Python finds the /app folder
+ENV PYTHONPATH=/app
+
+# 6. Expose port
 EXPOSE 7860
 
-# 8. Start the node
+# 7. Start the node - using the full path to be safe
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]