Update app.py

app.py

@@ -47,10 +47,10 @@ def get_similar_quotes(philosopher, question):
   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, -3)[-3:]
+  ind = np.argpartition(sims, -4)[-4:]
   similar_sentences = [df['quote'][i] for i in ind]
-  top3quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,4))
-  top3quotes['Quotes'] = top3quotes['Quotes'].str[:-1].str[:250] + "..."
+  top4quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,5))
+  top4quotes['Quotes'] = top4quotes['Quotes'].str[:-1].str[:250] + "..."
   return top3quotes
 
 def main(question, philosopher):