Update Dockerfile

Dockerfile

@@ -1,15 +1,18 @@
-FROM python:3.9-slim
+# Use an official Python runtime as a parent image
+FROM python:3.10-slim
 
-WORKDIR /app
+# Set the working directory in the container
+WORKDIR /code
 
-# Install dependencies
-RUN pip install torch transformers fastapi uvicorn pydantic
+# Copy the requirements file into the container at /code
+COPY ./requirements.txt /code/requirements.txt
 
-# Copy model loading script
-COPY app.py .
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# Expose port
-EXPOSE 7860
+# Copy the rest of the application code into the container at /code
+COPY ./main.py /code/main.py
 
-# Run the API
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Command to run the app using uvicorn.
+# It will be available on port 7860, the standard for HF Spaces
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]