Update app.py

app.py

@@ -376,12 +376,22 @@ def server(input, output, session):
             'x0',
             'z0']
         
+        
+        # Assuming `df` is your DataFrame and `features_table` contains column names as strings.
+        float_columns = [col for col in df.columns if df[col].dtype in [pl.Float32, pl.Float64]]
+
+        
         selection = [['pitcher_id','pitcher_name','batter_id','batter_name','pitcher_hand',
                       'batter_hand','balls','strikes','play_code','event_type','pitch_type','vaa','haa']+features_table+['tj_stuff_plus']]
 
-        
+
+# Applying the rounding only to float columns
+        rounded_selection = [
+            df[col].round(1) if col in float_columns else df[col] for col in selection
+        ]
+
         return render.DataGrid(
-                df.select(selection).to_pandas().round(1),
+                df.select(rounded_selection),
                 row_selection_mode='multiple',
                 height=None,
                 width='fit-content',