Update app.py

app.py

@@ -23,11 +23,12 @@ dice_types = {
 # Create a function to build the rolls pandas series to be a list of dictionaries
 def build_rolls(dice_type, num_rolls):
     sides = dice_types[dice_type]["sides"]
-    rolls = pd.Series([sides] * num_rolls).apply(pd.Series.sample, replace=True)
+    rolls = pd.Series([sides] * num_rolls).apply(pd.Series.sample, replace=True, axis=1)
     rolls = rolls.tolist()
     count = sum(rolls)
     return [{"Roll": roll, "Count": count, "DiceNumberOfSides": sides} for roll in rolls]
 
+
 # Create a selectbox to choose the dice type
 dice_type = st.selectbox("🎲 Select a dice type", options=list(dice_types.keys()), format_func=lambda x: x.split(' ')[-2])