melk2025 commited on
Commit
912c722
·
verified ·
1 Parent(s): 46ae155

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -43,7 +43,9 @@ collection = client.get_or_create_collection(
43
  )
44
 
45
  # Load the embedding model
46
- embedding_model = SentenceTransformer('sentence-transformers/paraphrase-MiniLM-L6-v2')
 
 
47
 
48
  # Initialize the text splitter
49
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1200, chunk_overlap=300)
@@ -79,7 +81,9 @@ SIMILARITY_THRESHOLD = 0.75
79
  client1 = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY) # Replace with your OpenRouter API key
80
 
81
  # ---------------------- Models ----------------------
82
- semantic_model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
 
 
83
 
84
  # Load QA Data
85
  with open("qa.json", "r", encoding="utf-8") as f:
 
43
  )
44
 
45
  # Load the embedding model
46
+ #embedding_model = SentenceTransformer('sentence-transformers/paraphrase-MiniLM-L6-v2')
47
+ embedding_model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
48
+
49
 
50
  # Initialize the text splitter
51
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1200, chunk_overlap=300)
 
81
  client1 = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY) # Replace with your OpenRouter API key
82
 
83
  # ---------------------- Models ----------------------
84
+ #semantic_model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
85
+
86
+ semantic_model = SentenceTransformer("sentence-transformers//paraphrase-multilingual-mpnet-base-v2")
87
 
88
  # Load QA Data
89
  with open("qa.json", "r", encoding="utf-8") as f: