Update dockerfile

dockerfile

@@ -1,27 +1,17 @@
-# Declare your environment variables with the ARG directive
-ARG MODEL_REPO_NAME
-
-# Use the official Python image
+# Use official Python runtime
 FROM python:3.9
 
-# Set the working directory inside the container
+# Set the working directory
 WORKDIR /app
 
 # Copy application files
 COPY . .
 
-# Install required dependencies
+# Install 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 the correct port for FastAPI
 EXPOSE 7860
 
-# Set Flask environment variables
-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"]
+# Run the FastAPI app with Uvicorn
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]