singhankur01 commited on
Commit
0fb8c62
·
verified ·
1 Parent(s): d307c53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -29,7 +29,18 @@ from langchain.retrievers.document_compressors import CrossEncoderReranker
29
  from langchain_community.cross_encoders import HuggingFaceCrossEncoder
30
  from langchain.prompts import ChatPromptTemplate
31
 
 
 
 
 
32
 
 
 
 
 
 
 
 
33
  # Load environment variables
34
  load_dotenv()
35
 
 
29
  from langchain_community.cross_encoders import HuggingFaceCrossEncoder
30
  from langchain.prompts import ChatPromptTemplate
31
 
32
+ import os
33
+ from sentence_transformers import SentenceTransformer
34
+
35
+ MODEL_DIR = "models/all-MiniLM-L6-v2"
36
 
37
+ if not os.path.exists(MODEL_DIR):
38
+ print("📦 Downloading SentenceTransformer model...")
39
+ model = SentenceTransformer("all-MiniLM-L6-v2")
40
+ model.save(MODEL_DIR)
41
+ print("✅ Model saved at", MODEL_DIR)
42
+
43
+
44
  # Load environment variables
45
  load_dotenv()
46