Shreyas094 commited on
Commit
f800b56
·
verified ·
1 Parent(s): 1dd8b2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -149,7 +149,7 @@ def summarize_article(article, content, model, system_prompt, user_query, client
149
  Article:
150
  Title: {article['title']}
151
  URL: {article['href']}
152
- Content: {article['body'][:500]}... # Truncate to avoid extremely long prompts
153
 
154
  Additional Context:
155
  {content[:1000]}... # Truncate additional context as well
@@ -194,9 +194,12 @@ def get_response_with_search(query, model, num_calls=3, temperature=0.2, use_emb
194
  search_results = duckduckgo_search(query)
195
  client = InferenceClient(model, token=huggingface_token)
196
 
 
 
 
197
  summaries = []
198
  for result in search_results:
199
- summary = summarize_article(result, model, system_prompt, query, client, temperature)
200
  summaries.append({
201
  "title": result['title'],
202
  "url": result['href'],
 
149
  Article:
150
  Title: {article['title']}
151
  URL: {article['href']}
152
+ Content: {article['body'][:1000]}... # Truncate to avoid extremely long prompts
153
 
154
  Additional Context:
155
  {content[:1000]}... # Truncate additional context as well
 
194
  search_results = duckduckgo_search(query)
195
  client = InferenceClient(model, token=huggingface_token)
196
 
197
+ # Prepare overall context
198
+ overall_context = "\n".join([f"{result['title']}\n{result['body']}" for result in search_results])
199
+
200
  summaries = []
201
  for result in search_results:
202
+ summary = summarize_article(result, overall_context, model, system_prompt, query, client, temperature)
203
  summaries.append({
204
  "title": result['title'],
205
  "url": result['href'],