Update app.py

app.py

@@ -54,7 +54,7 @@ if query:
     st.subheader(f"Similar Words to {query}")
 
     # calculate the sizes of the squares in the treemap
-    short_table = table.head(20)
+    short_table = table.head(10)
     short_table.index += 1
     short_table.index = 1 / short_table.index
     sizes = short_table.index.tolist()
@@ -63,7 +63,7 @@ if query:
     color = [cmap[i] for i in range(len(sizes))]
 
     short_table.set_index('Word', inplace=True)
-    squarify.plot(sizes=sizes, label=short_table.index.tolist(), color=color, edgecolor="#EBF5FB", text_kwargs={'fontsize': 6})
+    squarify.plot(sizes=sizes, label=short_table.index.tolist(), color=color, edgecolor="#EBF5FB", text_kwargs={'fontsize': 12})
     # # plot the treemap using matplotlib
     plt.axis('off')
     fig = plt.gcf()
@@ -89,7 +89,7 @@ if query:
     # time.sleep(2)
     st.subheader(f"Similar Genes to {query}")
 
-    df1 = df1.head(20)
+    df1 = df1.head(10)
     df1.index = 1/df1.index
     sizes = df1.index.tolist()
 
@@ -97,7 +97,7 @@ if query:
     color2 = [cmap2[i] for i in range(len(sizes))]
 
     df1.set_index('Human Gene', inplace=True)
-    squarify.plot(sizes=sizes, label=df1.index.tolist(), color=color2, edgecolor="#EBF5FB", text_kwargs={'fontsize': 8})
+    squarify.plot(sizes=sizes, label=df1.index.tolist(), color=color2, edgecolor="#EBF5FB", text_kwargs={'fontsize': 12})
     #
     # # plot the treemap using matplotlib