Create Dockerfile

Dockerfile

@@ -0,0 +1,22 @@
+# Use an official Python image
+FROM python:3.10
+
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/data/.cache
+ENV HF_HOME=/data/.cache
+
+# Install dependencies
+RUN pip install --upgrade pip
+RUN pip install fastapi uvicorn transformers torch
+
+# Create app directory
+WORKDIR /app
+
+# Copy code
+COPY main.py .
+
+# Expose port
+EXPOSE 7860
+
+# Start FastAPI
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]