Spaces:
Sleeping
Sleeping
Commit ·
a00b302
1
Parent(s): 6fa786c
Updated
Browse files- Dockerfile +3 -1
- app.py +1 -1
- requirements.txt +2 -1
- 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
|
|
|
|
|
|
|
| 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
|
| 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
|
| 3 |
-
from
|
|
|
|
| 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"
|