Fixing requirements

Dockerfile

@@ -4,10 +4,7 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install -r --no-cache-dir requirements.txt
 # preload models
-RUN python -c '\
-from transformers import BartForConditionalGeneration, AutoTokenizer;\
-AutoTokenizer.from_pretrained("ibm/materials.selfies-ted");\
-BartForConditionalGeneration.from_pretrained("ibm/materials.selfies-ted")'
+
 COPY . .
 
 CMD ["python", "app.py"]

requirements.txt

@@ -2,7 +2,6 @@
 wheel
     torch>=2.1.0            
     transformers>=4.40.0
-    pytorch-fast-transformers==0.4.0
     regex                   
     numpy==1.26.4        
     pandas==1.4.0