Update app.py

app.py

@@ -101,7 +101,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
     gr.Markdown("""<h1><center>Aira Demo (Português) 🤓💬</h1></center>""")
     gr.Markdown(intro)
 
-    chatbot = gr.Chatbot(label="Aira").style(height=500)
+    chatbot = gr.Chatbot(label="Aira", height=500)
     msg = gr.Textbox(label="Escreva uma pergunta ou instrução para Aira ...", placeholder="Olá Aira, como vai você?")
 
     # Parameters to control the generation
@@ -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
         )
@@ -225,14 +224,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)
 
     
     
@@ -245,4 +244,4 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
     submit.click(fn=search_in_datset, inputs=[search_field, search_input], outputs=out_dataframe)
 
 demo.queue()
-demo.launch()
+demo.launch()