Shreyas094 commited on
Commit
27b795a
1 Parent(s): 201ffe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -232,16 +232,18 @@ def summarize_content(content, model):
232
  return summary
233
 
234
  def rank_search_results(titles, summaries, model):
235
- ranking_prompt = f"""
236
- Rank the following search results from a financial analyst perspective.
237
- Assign a rank from 1 to {len(titles)} based on relevance, with 1 being the most relevant.
238
- Return only the numeric ranks in order, separated by commas.
239
-
240
- Titles and summaries:
241
- {', '.join([f"{i+1}. Title: {title}\nSummary: {summary}" for i, (title, summary) in enumerate(zip(titles, summaries))])}
242
-
243
- Ranks:
244
- """
 
 
245
  ranks_str = generate_chunked_response(model, ranking_prompt)
246
  ranks = [float(rank.strip()) for rank in ranks_str.split(',')]
247
  return ranks
 
232
  return summary
233
 
234
  def rank_search_results(titles, summaries, model):
235
+ ranking_prompt = (
236
+ "Rank the following search results from a financial analyst perspective. "
237
+ f"Assign a rank from 1 to {len(titles)} based on relevance, with 1 being the most relevant. "
238
+ "Return only the numeric ranks in order, separated by commas.\n\n"
239
+ "Titles and summaries:\n"
240
+ )
241
+
242
+ for i, (title, summary) in enumerate(zip(titles, summaries), 1):
243
+ ranking_prompt += f"{i}. Title: {title}\nSummary: {summary}\n\n"
244
+
245
+ ranking_prompt += "Ranks:"
246
+
247
  ranks_str = generate_chunked_response(model, ranking_prompt)
248
  ranks = [float(rank.strip()) for rank in ranks_str.split(',')]
249
  return ranks