Update app.py

app.py

@@ -1180,7 +1180,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
                 # Filter the DataFrame to include only the required columns
                 #print(df[required_columns])
                 #limited_data = df[required_columns]
-                limited_data = limited_data.head(3)
+                limited_data11 = df.head(3)
+                limited_data = limited_data11[required_columns]
                 limited_data_without_first_column = limited_data.iloc[:, 1:] 
         else:
                 limited_data = df.head(20)