Add application file

Dockerfile

@@ -1,9 +1,23 @@
+# Use an official Python runtime as a parent image
 FROM python:3.10-slim
 
+# Set the working directory in the container
 WORKDIR /app
 
-COPY ./ /app
+# Copy the current directory contents into the container at /app
+COPY . /app
 
-RUN pip install -r requirements.txt
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Make port 80 available to the world outside this container
+EXPOSE 80
+
+# Define environment variable
+ENV TRANSFORMERS_CACHE=/app/.cache
+
+# Create cache directory
+RUN mkdir -p /app/.cache
+
+# Run app.py when the container launches
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]

requirements.txt

@@ -1,10 +1,5 @@
-fastapi[standard]
-uvicorn[standard]
-transformers[standard]
-torch[standard]
-torchvision[standard]
-torchaudio[standard]
 fastapi
 uvicorn
 transformers
-scipy
+torch
+scipy