fix type return

main.py

@@ -70,7 +70,7 @@ def get_timers(
     return timers
 
 
-def get_plot(timers: Dict[str, Union[float, List[float]]]) -> plotly.graph_objs._figure.Figure:
+def get_plot(timers: Dict[str, Union[float, List[float]]]) -> plotly.graph_objs.Figure:
     """
     Plot the inference time for each model.