alaselababatunde commited on
Commit
a00b302
·
1 Parent(s): 6fa786c
Files changed (4) hide show
  1. Dockerfile +3 -1
  2. app.py +1 -1
  3. requirements.txt +2 -1
  4. vector.py +3 -2
Dockerfile CHANGED
@@ -37,7 +37,9 @@ EXPOSE 7860
37
 
38
  # Hugging Face cache environment variables
39
  ENV HF_HOME=/app/huggingface_cache
40
- ENV TRANSFORMERS_CACHE=/app/huggingface_cache
 
 
41
 
42
  # Start FastAPI app on port 7860
43
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
37
 
38
  # Hugging Face cache environment variables
39
  ENV HF_HOME=/app/huggingface_cache
40
+ ENV HF_DATASETS_CACHE=/app/huggingface_cache
41
+ ENV TORCH_HOME=/app/huggingface_cache
42
+
43
 
44
  # Start FastAPI app on port 7860
45
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from fastapi import FastAPI, UploadFile, File
3
  from langchain.prompts import PromptTemplate
4
- from langchain_huggingface import HuggingFaceEndpoint
5
  from vector import query_vector
6
 
7
  app = FastAPI(title="AgriCopilot")
 
1
  import os
2
  from fastapi import FastAPI, UploadFile, File
3
  from langchain.prompts import PromptTemplate
4
+ from langchain_huggingface import HuggingFaceEmbeddings
5
  from vector import query_vector
6
 
7
  app = FastAPI(title="AgriCopilot")
requirements.txt CHANGED
@@ -6,4 +6,5 @@ huggingface-hub
6
  sentence-transformers
7
  langchain_huggingface
8
  langchain_community
9
- langchain
 
 
6
  sentence-transformers
7
  langchain_huggingface
8
  langchain_community
9
+ langchain
10
+ langchain-huggingface
vector.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
- from langchain.vectorstores import FAISS
3
- from langchain.embeddings import HuggingFaceEmbeddings
 
4
 
5
  # ----------------- CONFIG -----------------
6
  VECTOR_PATH = "faiss_index"
 
1
  import os
2
+ from langchain_community.vectorstores import FAISS
3
+ from langchain_community.embeddings import HuggingFaceEmbeddings
4
+
5
 
6
  # ----------------- CONFIG -----------------
7
  VECTOR_PATH = "faiss_index"