How can I style the color bar scale and set the figure with the size?

Hey fellows,

I’m trying to create a map chart with 640 pixels, but the figure is very smaller than the “color bar scale”

def map_summary_cards(data, isin_vals = [None]):

    fig = px.choropleth(locations=data['state_code'],
                    color=data['Total'].astype(int),
                    locationmode="USA-states", 
                    scope='usa', color_continuous_scale=px.colors.sequential.Viridis)

    fig.update_layout(margin=dict(t=50, b=0, r=15, l=15),
                      title="MAP US", title_x=.5,
                      height=640)
    return fig    

fig_map = map_summary_cards(map_data)

I want to set the map in a size that fits with the application and also, I don’t want the color bar too large like it is

Could someone give me some help on it?