Update Dockerfile

Dockerfile

@@ -1,2 +1,17 @@
 FROM huggingface/autotrain-advanced:latest
 CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
+
+# Base image with AutoTrain
+FROM huggingface/autotrain-advanced:latest
+
+# Set working directory
+WORKDIR /app
+
+# Copy all files from Hugging Face Space
+COPY . /app/
+
+# Install dependencies (if needed)
+RUN pip install --no-cache-dir pandas faiss-cpu requests numpy
+
+# Run training script
+CMD ["bash", "train.sh"]