Update app.py

app.py

@@ -1233,10 +1233,11 @@ def handle_large_dataset(df, create_document,isDataFrame):
                 limited_data_top3 = df.head(3)
                 if len(df.columns) > 5:
                     limited_data = limited_data_top3[required_columns]
+                    limited_data_without_first_column = limited_data.iloc[:, 1:] 
                 else:
-                    limited_data = limited_data_top3
+                    limited_data_without_first_column = limited_data_top3
+                
                 
-                limited_data_without_first_column = limited_data.iloc[:, 1:] 
         else:
                 limited_data = df.head(20)
                 limited_data_without_first_column = limited_data