Create Dockerfile

Dockerfile

@@ -0,0 +1,29 @@
+# Use an official PyTorch image as a base (includes CUDA for GPU support if available)
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+
+# Set the working directory inside the container
+WORKDIR /app
+
+# Install Python dependencies
+RUN pip install --no-cache-dir \
+    pandas \
+    torch \
+    sentence-transformers \
+    transformers \
+    numpy \
+    faiss-cpu \
+    gunicorn \
+    huggingface_hub
+
+# Copy the necessary files into the container
+COPY plant_data_chunks_and_embeddings.csv /app/plant_data_chunks_and_embeddings.csv
+COPY main.py /app/main.py
+
+# Set the environment variable to point to the app directory
+ENV PYTHONPATH /app
+
+# Expose the port the app will run on (useful when you run the app in the container)
+EXPOSE 5000
+
+# Command to run the app (Gunicorn is a WSGI server that serves the Flask app)
+CMD ["gunicorn", "--bind", "0.0.0.0:5000", "main:app"]