From 520a0bbd3418d38c755be7c2ad4db95c3fa95f2c Mon Sep 17 00:00:00 2001 From: nlpfun Date: Mon, 18 Apr 2022 23:10:51 +0800 Subject: [PATCH] Update README --- README.md | 4 +++- requirements.txt | 11 ++++------- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index af0c334..6a8f7a3 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,9 @@ Alignment between 25 languages: Catalan (ca), Chinese (zh), Czech (cs), Danish ( ## Installation -Please see [requirements.txt](./requirements.txt) for installation information. If you are running Bertalign on *GPU-enabled Linux* such as Google Colaboratory, please install *faiss-gpu* for faster processing. +Please see [requirements.txt](./requirements.txt) for installation information. + +### Run Bertalign directly in a [Google Colab notebook](https://colab.research.google.com/drive/123GhXwgwmQp1F5SVZ74_uIgyxo6hLRq0?usp=sharing) ## Basic example diff --git a/requirements.txt b/requirements.txt index 58156f7..fe5e7f5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,5 @@ -faiss_cpu==1.7.2 -#faiss_gpu==1.7.2 +faiss-gpu==1.7.2 +#faiss-cpu==1.7.2 googletrans==4.0.0rc1 -numba==0.55.1 -numpy==1.21.6 -sentence_splitter==1.4 -sentence_transformers==2.2.0 -torch==1.11.0+cu113 +sentence-splitter==1.4 +sentence-transformers==2.2.0