diff --git a/README.md b/README.md
index aa34310..ae20ff1 100644
--- a/README.md
+++ b/README.md
@@ -1,10 +1,16 @@
# MiniGPT-4 and MiniGPT-v2
+**MiniGPT-v2: Large Language Model as a Unified Interface for Vision-Language Multi-task Learning**
-**King Abdullah University of Science and Technology**
+
+
+
+**MiniGPT-4: Enhancing Vision-language Understanding with Advanced Large Language Models**
[](https://colab.research.google.com/drive/1OK4kYsZphwt5DXchKkzMBjYF6jnkqh4R?usp=sharing) [](https://www.youtube.com/watch?v=__tftoxpBAw&feature=youtu.be)
+*King Abdullah University of Science and Technology*
+
## 💡 Get help - [Q&A](https://github.com/Vision-CAIR/MiniGPT-4/discussions/categories/q-a) or [Discord 💬](https://discord.gg/5WdJkjbAeE)
diff --git a/environment.yml b/environment.yml
index c17288d..cf90e89 100644
--- a/environment.yml
+++ b/environment.yml
@@ -7,56 +7,27 @@ dependencies:
- python=3.9
- cudatoolkit
- pip
- - pytorch=2.0.0
- - pytorch-mutex=1.0=cuda
- - torchaudio=0.12.1
- - torchvision=0.13.1
- pip:
- - accelerate==0.20.3
- - aiohttp==3.8.4
- - aiosignal==1.3.1
- - async-timeout==4.0.2
- - attrs==22.2.0
- - bitsandbytes==0.37.0
- - cchardet==2.1.7
- - chardet==5.1.0
- - contourpy==1.0.7
- - cycler==0.11.0
- - filelock==3.9.0
- - fonttools==4.38.0
- - frozenlist==1.3.3
+ - torch==2.0.0
+ - torchaudio
+ - torchvision
- huggingface-hub==0.18.0
- - importlib-resources==5.12.0
- - kiwisolver==1.4.4
- matplotlib==3.7.0
- - multidict==6.0.4
- - openai==0.27.0
- - packaging==23.0
- psutil==5.9.4
- - pycocotools==2.0.6
- - pyparsing==3.0.9
- - python-dateutil==2.8.2
+ - iopath
- pyyaml==6.0
- regex==2022.10.31
- tokenizers==0.13.2
- tqdm==4.64.1
- - transformers==4.32.0
+ - transformers==4.30.0
- timm==0.6.13
- - spacy==3.5.1
- webdataset==0.2.48
- - scikit-learn==1.2.2
- - scipy==1.10.1
- - yarl==1.8.2
- - zipp==3.14.0
- omegaconf==2.3.0
- opencv-python==4.7.0.72
- - iopath==0.1.10
- decord==0.6.0
- - tenacity==8.2.2
- peft==0.2.0
- - pycocoevalcap
- sentence-transformers
- - umap-learn
- - notebook
- gradio==3.47.1
+ - accelerate==0.20.3
+ - bitsandbytes==0.37.0
- wandb