Update dockerfile

dockerfile

@@ -1,3 +1,6 @@
+# Declare your environment variables with the ARG directive
+ARG MODEL_REPO_NAME
+
 # Use the official Python image
 FROM python:3.9
 
@@ -10,6 +13,9 @@ COPY . .
 # Install required dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Use the build-time variable in the command
+RUN python predict.py $MODEL_REPO_NAME
+
 # Expose the correct port for Hugging Face Spaces
 EXPOSE 7860
 
@@ -18,4 +24,4 @@ ENV FLASK_APP=app.py
 ENV FLASK_RUN_HOST=0.0.0.0
 
 # Run the Flask application on port 7860
-CMD ["flask", "run", "--host=0.0.0.0", "--port=7860"]
+CMD ["flask", "run", "--host=0.0.0.0", "--port=7860"]