Update app.py

app.py

@@ -506,7 +506,7 @@ if uploaded_file is not None:
         df['country'] = df['country'].str.strip()
         df = df.dropna(subset=['country', 'poem'])
         # Extract 10 rows per country
-        sampled_df = df.groupby('country').apply(lambda x: x.head(50)).reset_index(drop=True)
+        sampled_df = df.groupby('country').apply(lambda x: x.head(20)).reset_index(drop=True)
         
         st.subheader("Topic Modeling Settings")
         col1, col2 = st.columns(2)