fix plotly figure and colors

main.py

@@ -7,6 +7,7 @@ Author:
 """
 
 import plotly
+import plotly.figure_factory as ff
 import numpy as np
 import pandas as pd
 import streamlit as st
@@ -80,10 +81,8 @@ def get_plot(timers: Dict[str, Union[float, List[float]]]) -> plotly.graph_objs.
         Dictionary of inference times for each model.
     """
     data = pd.DataFrame.from_dict(timers, orient="columns")
-    colors = ["#140f0d", "#2b2c4f", "#615aa2", "#a991fa"]
-    fig = plotly.figure_factory.create_distplot(
-        [data[col] for col in data.columns], data.columns, bin_size=0.2, colors=colors
-    )
+    colors = ["#84353f", "#b4524b", "#f47e58", "#ffbe67"]
+    fig = ff.create_distplot([data[col] for col in data.columns], data.columns, bin_size=0.2, colors=colors)
     fig.update_layout(title_text="Inference Time", xaxis_title="Inference Time (s)", yaxis_title="Number of Samples")
     return fig