Update app.py

app.py

@@ -43,15 +43,15 @@ def ask_philosopher(philosopher, question):
   return answer
 
 def get_similar_quotes(philosopher, question):
-  """ Return top 3 most similar quotes to the question from a philosopher's dataframe """
+  """ Return top 5 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, -4)[-4:]
+  ind = np.argpartition(sims, -5)[-5:]
   similar_sentences = [df['quote'][i] for i in ind]
-  top4quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,5))
-  top4quotes['Quotes'] = top4quotes['Quotes'].str[:-1].str[:250] + "..."
-  return top4quotes
+  top5quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,6))
+  top5quotes['Quotes'] = top4quotes['Quotes'].str[:-1].str[:250] + "..."
+  return top5quotes
 
 def main(question, philosopher):
   out_image = "marcus-aurelius.jpg"
@@ -76,7 +76,7 @@ with gr.Blocks(css=".gradio-container {background-image: url('file=mountains_res
                 )
           
     with gr.Row():
-        out_image = gr.Image(label="Picture")
+        out_image = gr.Image(label="Picture", image_mode="L")
         out2 = gr.DataFrame(
                     headers=["Quotes"],
                     max_rows=5,