thechaiexperiment commited on
Commit
f7bc8e0
·
verified ·
1 Parent(s): 470fac8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -523,7 +523,8 @@ async def chat_endpoint(chat_query: ChatQuery):
523
  language_code = chat_query.language_code
524
  if language_code == 0:
525
  query_text = translate_ar_to_en(query_text)
526
- query_embedding = embed_query_text(query_text)
 
527
  embeddings_data = load_embeddings ()
528
  folder_path = 'downloaded_articles/downloaded_articles'
529
  initial_results = query_embeddings(query_embedding, embeddings_data, n_results)
 
523
  language_code = chat_query.language_code
524
  if language_code == 0:
525
  query_text = translate_ar_to_en(query_text)
526
+ query_embedding = embed_query_text(query_text)
527
+ n_results = 5
528
  embeddings_data = load_embeddings ()
529
  folder_path = 'downloaded_articles/downloaded_articles'
530
  initial_results = query_embeddings(query_embedding, embeddings_data, n_results)