Fix: Using PyTorch base image and updating requirements

Dockerfile

@@ -0,0 +1,12 @@
+FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
+
+WORKDIR /app
+
+COPY requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir fastapi uvicorn[standard] pydantic transformers tweepy numpy
+
+COPY . /app
+
+ENV PORT 7860
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -2,7 +2,6 @@ fastapi
 uvicorn[standard]
 python-multipart
 transformers
-torch
 pandas
 tweepy 
 emoji