Update Dockerfile

Dockerfile

@@ -4,8 +4,18 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# 4. Install dependencies
+RUN pip install -r requirements.txt
 
-COPY . .
+# 5. Optionally install TensorFlow for GPU support (if available)
+RUN if [ $(command -v nvidia-smi) ]; then pip install tensorflow-gpu; else pip install tensorflow; fi
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+# 6. Copy application code
+COPY index.py ./
+
+# 7. Expose port for Flask app
+EXPOSE 5000
+
+# 8. Start the Flask app
+CMD ["python", "index.py"]
+CMD ["uvicorn", "index:app", "--host", "0.0.0.0", "--port", "7860"]