Deepti-AI commited on
Commit
a4913cf
·
verified ·
1 Parent(s): 0ca5df5

Update main.py

Browse files

the retrieved chunks are changed to telugu from english

Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -204,7 +204,8 @@ async def retrieve_top_k_hybrid(query, k, sem_weight,syn_weight,bm25):
204
 
205
  top_results = df_expanded.sort_values(by="similarity", ascending=False).head(k)
206
 
207
- return top_results["Chunks"].to_list()
 
208
 
209
 
210
  os.makedirs("/tmp/transformers_cache", exist_ok=True)
@@ -332,7 +333,7 @@ def tts_chunk_stream(text_chunk: str, lang: str = "en"):
332
  async def get_rag_response(user_message: str):
333
  global chat_messages
334
  Chunks = await retrieve_top_k_hybrid(user_message,15, 0.9, 0.1,bm25)
335
- context = "======================================================================================================\n".join(Chunks)
336
  chat_messages.append({"role": "user", "content": f'''
337
  Context : {context}
338
  User Query: {user_message}'''})
 
204
 
205
  top_results = df_expanded.sort_values(by="similarity", ascending=False).head(k)
206
 
207
+ # return top_results["Chunks"].to_list()
208
+ return top_results["telugu_chunk"].to_list()
209
 
210
 
211
  os.makedirs("/tmp/transformers_cache", exist_ok=True)
 
333
  async def get_rag_response(user_message: str):
334
  global chat_messages
335
  Chunks = await retrieve_top_k_hybrid(user_message,15, 0.9, 0.1,bm25)
336
+ context = "======================================================================================================\n".join(map(str,Chunks))
337
  chat_messages.append({"role": "user", "content": f'''
338
  Context : {context}
339
  User Query: {user_message}'''})