DaudiAI commited on
Commit
66367c8
·
1 Parent(s): 797ab7f

add Dockerfile and HF deployment config

Browse files
Files changed (3) hide show
  1. Dockerfile +12 -0
  2. requirements.txt +3 -2
  3. ui/app.py +10 -1
Dockerfile ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.11-slim
2
+
3
+ WORKDIR /app
4
+
5
+ COPY requirements.txt .
6
+ RUN pip install -r requirements.txt
7
+
8
+ COPY . .
9
+
10
+ EXPOSE 7860
11
+
12
+ CMD ["streamlit", "run", "ui/app.py", "--server.port", "7860", "--server.address", "0.0.0.0"]
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  requests
2
  beautifulsoup4
3
  lxml
4
- chromadb
5
- sentence-transformers
6
  openai
7
  streamlit
 
 
1
  requests
2
  beautifulsoup4
3
  lxml
4
+ chromadb==0.4.24
5
+ sentence-transformers==2.2.2
6
  openai
7
  streamlit
8
+ huggingface_hub
ui/app.py CHANGED
@@ -12,7 +12,16 @@ import chromadb
12
  from sentence_transformers import SentenceTransformer
13
  from openai import OpenAI
14
  import random
15
-
 
 
 
 
 
 
 
 
 
16
  # --- Config ---
17
  CHROMA_DIR = "data/chroma_db"
18
  COLLECTION_NAME = "kenya_law"
 
12
  from sentence_transformers import SentenceTransformer
13
  from openai import OpenAI
14
  import random
15
+ import os
16
+ from huggingface_hub import snapshot_download
17
+
18
+ if not os.path.exists("data/chroma_db"):
19
+ os.makedirs("data", exist_ok=True)
20
+ snapshot_download(
21
+ repo_id="Daudipdg/iroh-chroma-db",
22
+ repo_type="dataset",
23
+ local_dir="data/chroma_db"
24
+ )
25
  # --- Config ---
26
  CHROMA_DIR = "data/chroma_db"
27
  COLLECTION_NAME = "kenya_law"