Spaces:
Sleeping
Sleeping
DD update
Browse files
app.py
CHANGED
|
@@ -9,7 +9,6 @@ import os
|
|
| 9 |
import joblib
|
| 10 |
from openai import OpenAI
|
| 11 |
|
| 12 |
-
import os
|
| 13 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp"
|
| 14 |
os.environ["HF_HOME"] = "/tmp"
|
| 15 |
os.environ["XDG_CACHE_HOME"] = "/tmp"
|
|
@@ -34,7 +33,7 @@ def load_model_and_index(chunks):
|
|
| 34 |
embeddings = model.encode(chunks)
|
| 35 |
faiss_index = faiss.IndexFlatL2(embeddings.shape[1])
|
| 36 |
faiss_index.add(np.array(embeddings))
|
| 37 |
-
|
| 38 |
return model, chunks, faiss_index
|
| 39 |
|
| 40 |
def search(query, model, chunks, index, k=3):
|
|
|
|
| 9 |
import joblib
|
| 10 |
from openai import OpenAI
|
| 11 |
|
|
|
|
| 12 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp"
|
| 13 |
os.environ["HF_HOME"] = "/tmp"
|
| 14 |
os.environ["XDG_CACHE_HOME"] = "/tmp"
|
|
|
|
| 33 |
embeddings = model.encode(chunks)
|
| 34 |
faiss_index = faiss.IndexFlatL2(embeddings.shape[1])
|
| 35 |
faiss_index.add(np.array(embeddings))
|
| 36 |
+
# joblib.dump((model, chunks, faiss_index), "rag_model.joblib")
|
| 37 |
return model, chunks, faiss_index
|
| 38 |
|
| 39 |
def search(query, model, chunks, index, k=3):
|