Merge pull request #47 from minggnim/minor-updates

Minor updates

apps/README.md

@@ -0,0 +1,16 @@
+## Instructions to run applications
+
+1. Install the latest nlp_models package
+
+    ```
+    pip install -U nlp_models
+    ```
+
+2. Download Llama2 quantization model `llama-2-7b-chat.ggmlv3.q8_0.bin` 
+    
+    from https://huggingface.co/TheBloke/Llama-2-13B-chat-GGML/tree/main and place it under `models` folder
+
+
+3. Run the application of interest
+    - Chat `streamlit run chat.py`
+    - Q&A `streamlit run qa.py` 

apps/chat.py

@@ -1,14 +1,3 @@
-"""
-To run Llama2 chat UI on CPU
-1. Install the latest nlp_models package 
-`pip install -U nlp_models`
-2. Download `llama-2-7b-chat.ggmlv3.q8_0.bin` 
-from https://huggingface.co/TheBloke/Llama-2-13B-chat-GGML/tree/main 
-and place it under `models` folder
-3. Run `streamlit run chat.py` to spin up the UI
-"""
-
-
 import streamlit as st
 from langchain.memory import ConversationBufferWindowMemory
 from nlp_models.llm.base import LlmConfig

apps/qa.py

@@ -1,11 +1,3 @@
-"""
-To run Llama2 QA UI on CPU
-1. Install the latest nlp_models package `pip install -U nlp_models`
-2. Download `llama-2-7b-chat.ggmlv3.q8_0.bin` from https://huggingface.co/TheBloke/Llama-2-13B-chat-GGML/tree/main and place it under `models` folder
-3. Run `streamlit run qa.py` to spin up the UI
-"""
-
-
 import streamlit as st
 from tempfile import NamedTemporaryFile
 from nlp_models.llm.base import LlmConfig

requirements.txt

@@ -11,5 +11,6 @@ fastapi
 
 watchdog
 pypdf
+pypdfium2
 sentence_transformers
 faiss-cpu

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = nlp-models
-version = 4.2.0
+version = 4.2.1
 author= Ming Gao
 author_email = [email protected]
 url = https://github.com/minggnim/nlp-models