Update Dockerfile

Dockerfile

@@ -1,10 +1,14 @@
+# Use Hugging Face autotrain base image
 FROM huggingface/autotrain-advanced:latest
 
-# Upgrade autotrain-advanced at build time
-RUN pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced
+# Upgrade pip just in case
+RUN pip install --upgrade pip
 
-# Expose port
+# Upgrade autotrain-advanced safely (without uninstalling)
+RUN pip install --upgrade autotrain-advanced
+
+# Expose the port for the app
 EXPOSE 7860
 
-# Start the app at runtime
+# Run the autotrain app
 CMD ["autotrain", "app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]