Théo ALVES DA COSTA commited on
Commit
393b23a
1 Parent(s): eead5d8

Updated reqs

Browse files
Files changed (5) hide show
  1. app.py +1 -1
  2. climateqa/engine/retriever.py +1 -1
  3. requirements.txt +2 -1
  4. setup.py +1 -0
  5. test +32 -0
app.py CHANGED
@@ -144,7 +144,7 @@ async def chat(query,history,audience,sources,reports):
144
  reports = []
145
 
146
  llm = get_llm(max_tokens = 1024,temperature = 0.0)
147
- retriever = ClimateQARetriever(vectorstore=vectorstore,sources = sources,reports = reports,k_summary = 3,k_total = 10,threshold=0.4)
148
  rag_chain = make_rag_chain(retriever,llm)
149
 
150
  source_string = ""
144
  reports = []
145
 
146
  llm = get_llm(max_tokens = 1024,temperature = 0.0)
147
+ retriever = ClimateQARetriever(vectorstore=vectorstore,sources = sources,reports = reports,k_summary = 3,k_total = 10,threshold=0.7)
148
  rag_chain = make_rag_chain(retriever,llm)
149
 
150
  source_string = ""
climateqa/engine/retriever.py CHANGED
@@ -13,7 +13,7 @@ class ClimateQARetriever(BaseRetriever):
13
  vectorstore:VectorStore
14
  sources:list = ["IPCC","IPBES"]
15
  reports:list = []
16
- threshold:float = 0.4
17
  k_summary:int = 3
18
  k_total:int = 10
19
  namespace:str = "vectors"
13
  vectorstore:VectorStore
14
  sources:list = ["IPCC","IPBES"]
15
  reports:list = []
16
+ threshold:float = 0.6
17
  k_summary:int = 3
18
  k_total:int = 10
19
  namespace:str = "vectors"
requirements.txt CHANGED
@@ -5,4 +5,5 @@ azure-storage-blob
5
  python-dotenv==1.0.0
6
  langchain==0.0.333
7
  pinecone-client==2.2.4
8
- sentence-transformers==2.2.2
 
5
  python-dotenv==1.0.0
6
  langchain==0.0.333
7
  pinecone-client==2.2.4
8
+ sentence-transformers
9
+ huggingface-hub
setup.py ADDED
@@ -0,0 +1 @@
 
1
+ print("yoooooo")
test ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10
2
+
3
+ WORKDIR /src
4
+
5
+ COPY requirements.txt .
6
+
7
+ RUN pip install --no-cache-dir -r requirements.txt
8
+
9
+ # Set up a new user named "user" with user ID 1000
10
+ RUN useradd -m -u 1000 user
11
+ # Switch to the "user" user
12
+ USER user
13
+ # Set home to the user's home directory
14
+ ENV HOME=/home/user \
15
+ PATH=/home/user/.local/bin:$PATH \
16
+ PYTHONPATH=$HOME/app \
17
+ PYTHONUNBUFFERED=1 \
18
+ GRADIO_ALLOW_FLAGGING=never \
19
+ GRADIO_NUM_PORTS=1 \
20
+ GRADIO_SERVER_NAME=0.0.0.0 \
21
+ GRADIO_THEME=huggingface \
22
+ SYSTEM=spaces
23
+
24
+ # Set the working directory to the user's home directory
25
+ WORKDIR $HOME/app
26
+
27
+ # Copy the current directory contents into the container at $HOME/app setting the owner to the user
28
+ COPY --chown=user . $HOME/app
29
+
30
+ CMD ["python","setup.py"]
31
+
32
+ CMD ["python", "app.py"]