Refactor Dockerfile and requirements.txt to streamline torch installation

Dockerfile

@@ -10,6 +10,8 @@ WORKDIR /app
 
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
+RUN pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/test/cpu
+
 
 COPY --chown=user . /app
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,7 +1,6 @@
 fastapi==0.74.*
 requests==2.27.*
 sentencepiece==0.1.*
-torch==2.4.1+cpu --find-links https://download.pytorch.org/whl/cpu
 transformers==4.*
 uvicorn[standard]==0.17.*
 transformers~=4.44.0