Update Dockerfile and requirements.txt

Dockerfile

@@ -12,7 +12,7 @@ USER user
 # ENV HOME=/home/user \
 #     PATH=/home/user/.local/bin:/home/user/.cargo/bin:$PATH
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:/home/user/.cargo/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH
 
 # Set the working directory in the container
 WORKDIR $HOME/app
@@ -41,4 +41,4 @@ EXPOSE 80
 RUN /venv/bin/pip freeze > requirements.txt
 
 # Run train_llm.py when the container launches
-CMD ["/bin/bash", "-c", "source /venv/bin/activate && python train_llm.py"]
+CMD ["/bin/bash", "-c", "source /venv/bin/activate && python app.py"]

requirements.txt

@@ -0,0 +1,143 @@
+accelerate==0.25.0
+aiohttp==3.9.1
+aiosignal==1.3.1
+anyio==4.2.0
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+arrow==1.3.0
+asttokens==2.4.1
+async-lru==2.0.4
+attrs==23.1.0
+Babel==2.14.0
+beautifulsoup4==4.12.2
+bleach==6.1.0
+certifi==2023.11.17
+cffi==1.16.0
+charset-normalizer==3.3.2
+comm==0.2.0
+datasets==2.16.0
+debugpy==1.8.0
+decorator==5.1.1
+defusedxml==0.7.1
+dill==0.3.7
+docstring-parser==0.15
+executing==2.0.1
+fastjsonschema==2.19.0
+filelock==3.13.1
+fqdn==1.5.1
+frozenlist==1.4.1
+fsspec==2023.10.0
+huggingface-hub==0.20.1
+idna==3.6
+ipykernel==6.27.1
+ipython==8.19.0
+ipywidgets==8.1.1
+isoduration==20.11.0
+jedi==0.19.1
+Jinja2==3.1.2
+json5==0.9.14
+jsonpointer==2.4
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.2
+jupyter==1.0.0
+jupyter-console==6.6.3
+jupyter-events==0.9.0
+jupyter-lsp==2.2.1
+jupyter_client==8.6.0
+jupyter_core==5.5.1
+jupyter_server==2.12.1
+jupyter_server_terminals==0.5.0
+jupyterlab==4.0.9
+jupyterlab-widgets==3.0.9
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.25.2
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib-inline==0.1.6
+mdurl==0.1.2
+mistune==3.0.2
+mpmath==1.3.0
+multidict==6.0.4
+multiprocess==0.70.15
+nbclient==0.9.0
+nbconvert==7.13.1
+nbformat==5.9.2
+nest-asyncio==1.5.8
+networkx==3.2.1
+notebook==7.0.6
+notebook_shim==0.2.3
+numpy==1.26.2
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.101
+nvidia-nvtx-cu12==12.1.105
+overrides==7.4.0
+packaging==23.2
+pandas==2.1.4
+pandocfilters==1.5.0
+parso==0.8.3
+peft==0.7.1
+pexpect==4.9.0
+platformdirs==4.1.0
+prometheus-client==0.19.0
+prompt-toolkit==3.0.43
+psutil==5.9.7
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow==14.0.2
+pyarrow-hotfix==0.6
+pycparser==2.21
+Pygments==2.17.2
+python-dateutil==2.8.2
+python-json-logger==2.0.7
+pytz==2023.3.post1
+PyYAML==6.0.1
+pyzmq==25.1.2
+qtconsole==5.5.1
+QtPy==2.4.1
+referencing==0.32.0
+regex==2023.10.3
+requests==2.31.0
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rich==13.7.0
+rpds-py==0.15.2
+safetensors==0.4.1
+Send2Trash==1.8.2
+shtab==1.6.5
+six==1.16.0
+sniffio==1.3.0
+soupsieve==2.5
+stack-data==0.6.3
+sympy==1.12
+terminado==0.18.0
+tinycss2==1.2.1
+tokenizers==0.15.0
+torch==2.1.2
+tornado==6.4
+tqdm==4.66.1
+traitlets==5.14.0
+transformers==4.36.2
+triton==2.1.0
+trl==0.7.6
+types-python-dateutil==2.8.19.14
+typing_extensions==4.9.0
+tyro==0.6.1
+tzdata==2023.3
+uri-template==1.3.0
+urllib3==2.1.0
+wcwidth==0.2.12
+webcolors==1.13
+webencodings==0.5.1
+websocket-client==1.7.0
+widgetsnbextension==4.0.9
+xxhash==3.4.1
+yarl==1.9.4