Update Dockerfile

Dockerfile

@@ -1,10 +1,18 @@
-FROM tensorflow/tensorflow:2.10.0-gpu
+# Use a CPU-based Python image compatible with Hugging Face
+FROM python:3.8-slim
 
+# Set working directory
 WORKDIR /app
 
+# Copy and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy your app
 COPY app/ ./app/
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Expose the default port
+EXPOSE 7860
+
+# Run FastAPI with Uvicorn
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]