Update Dockerfile

Dockerfile

@@ -1,11 +1,8 @@
+# 🧠 Use the official AutoTrain Advanced base image
 FROM huggingface/autotrain-advanced:latest
 
-# Force reinstall compatible datasets version that still includes load_metric
-RUN pip install --force-reinstall datasets==2.7.0
+# 🚀 Expose required port for web UI
+EXPOSE 7860
 
-# (Optional) Pin autotrain-advanced to a compatible version that expects load_metric
-RUN pip install --force-reinstall autotrain-advanced==0.8.35
-
-# If you have a requirements.txt for other dependencies, copy and install them:
-# COPY requirements.txt .
-# RUN pip install -r requirements.txt
+# 🏃 Start the AutoTrain UI when the container launches
+CMD ["autotrain", "app", "start"]