thechaiexperiment commited on
Commit
98c07df
·
verified ·
1 Parent(s): 0e9c9a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -374,10 +374,6 @@ def extract_relevant_portions(query_embedding, top_documents, embeddings_data, m
374
 
375
  relevant_portions[doc_id] = sorted_sentences
376
 
377
- print(f"Extracted relevant portions for Document ID {doc_id} (Similarity: {similarity:.4f}):")
378
- for i, sentence in enumerate(sorted_sentences, start=1):
379
- print(f" Portion {i}: {sentence[:100]}...") # Print first 100 characters for preview
380
-
381
  return relevant_portions
382
 
383
  except Exception as e:
@@ -516,7 +512,7 @@ async def chat_endpoint(chat_query: ChatQuery):
516
  if language_code == 0:
517
  query_text = translate_ar_to_en(query_text)
518
  query_embedding = embed_query_text(query_text)
519
- n_results = 5
520
  embeddings_data = load_embeddings ()
521
  folder_path = 'downloaded_articles/downloaded_articles'
522
  initial_results = query_embeddings(query_embedding, embeddings_data, n_results)
 
374
 
375
  relevant_portions[doc_id] = sorted_sentences
376
 
 
 
 
 
377
  return relevant_portions
378
 
379
  except Exception as e:
 
512
  if language_code == 0:
513
  query_text = translate_ar_to_en(query_text)
514
  query_embedding = embed_query_text(query_text)
515
+ n_results = 3
516
  embeddings_data = load_embeddings ()
517
  folder_path = 'downloaded_articles/downloaded_articles'
518
  initial_results = query_embeddings(query_embedding, embeddings_data, n_results)