Update app.py

app.py

@@ -43,15 +43,15 @@ def ask_philosopher(philosopher, question):
   return answer
 
 def get_similar_quotes(philosopher, question):
-  """ Return top 5 most similar quotes to the question from a philosopher's dataframe """
+  """ Return top 3 most similar quotes to the question from a philosopher's dataframe """
   df = philosopher_dictionary[philosopher]['dataframe']
   question_embedding = model.encode(question)
   sims = [util.dot_score(question_embedding, quote_embedding) for quote_embedding in df['Embedding']]
-  ind = np.argpartition(sims, -5)[-5:]
+  ind = np.argpartition(sims, -3)[-3:]
   similar_sentences = [df['quote'][i] for i in ind]
-  top5quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,6))
-  top5quotes['Quotes'] = top5quotes['Quotes'].str[:-1].str[:250] + "..."
-  return top5quotes
+  top3quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,4))
+  top3quotes['Quotes'] = top3quotes['Quotes'].str[:-1].str[:250] + "..."
+  return top3quotes
 
 def main(question, philosopher):
   return ask_philosopher(philosopher, question), get_similar_quotes(philosopher, question)