Update app.py

app.py

@@ -127,9 +127,8 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
         out_dataframe = gr.Dataframe(
             headers=df.columns.tolist(),
             datatype=["str", "str", "str"],
-            max_rows=5,
+            row_count=10,
             col_count=(3, "fixed"),
-            overflow_row_behaviour='paginate',
             wrap=True,
             interactive=False
         )
@@ -226,14 +225,14 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
             cosine_similarities = cosine_similarity(prompt_tfidf_matrix, search_vector)
             temp_df['Cosine Similarity'] = cosine_similarities
             temp_df.sort_values('Cosine Similarity', ascending=False, inplace=True)
-            return temp_df.head()
+            return temp_df.head(10)
         
         elif column_name == 'Completion':
             search_vector = completion_tfidf_vectorizer.transform([search_string])
             cosine_similarities = cosine_similarity(completion_tfidf_matrix, search_vector)
             temp_df['Cosine Similarity'] = cosine_similarities
             temp_df.sort_values('Cosine Similarity', ascending=False, inplace=True)
-            return temp_df.head()
+            return temp_df.head(10)