flask と vue を使用した NMT 用の Web UI
- Linux(推薦) + GPU(推薦)
- GPUを使用する場合は、nvidiaドライバーとcudaまたはAMDドライバーとROCmを適切にインストールする、その上に。GPUのVRAMは4G以上そしてcuda のバージョンは 11 以降 ROCm のバージョンは 5 以降であるほうがいいと思う。
- nodejs 18+
- python >= 3.6
- ディープラーニングフレームワークはpytorch>= 1.10ほうがいいとおもう、インストールについてはpytorch (https://pytorch.org/get-started/locally)のこうしきペイジに参考してください。
cd frontend
npm install
cd ..
- モデルファイルを huggingfaceからダウンロードして、modelsにに入れおきます。モデルって例えばfacebookのnllb、そしてほがのはいいです。
# ディレクトリ構造:
- backend
- nmt
- models
- xxx1(あたらしいディレクトリそしてモデルファイルはそこにある。)
- xxx2(あたらしいディレクトリそしてモデルファイルはそこにある。)
- config.yaml.exampleに参考してconfig.yamlを編集します。
- run.pyを実行する。
conda create -n nmt_web_ui python=3.10 -y
conda activate nmt_web_ui
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia -y # optional
pip install -r requirements.txt
python run.py
最後、http://localhost:5000 にvue ページをアクセスできる。
cd frontend
npm run dev
cd frontend
npm run build