Update build_gradio_graph.py

build_gradio_graph.py

@@ -55,7 +55,7 @@ def call_predict(inference_dict, cols_order, numerical_columns, target_columns):
         print(y_pred)
         outputs = []
         for i in range(len(target_columns)):
-            outputs += [y_pred[i]]
+            outputs += [y_pred[i][0]]
             outputs += [10]
         # outputs += [fig]
 
@@ -138,6 +138,8 @@ def add_gradio_component(config_dict, component_key):
         new_component = gr.Plot(label=config_dict[component_key]["label"], type="matplotlib")
     elif config_dict[component_key]["comp_type"] == "Dataframe":
         new_component = gr.Dataframe(wrap=True, type="pandas")
+    elif config_dict[component_key]["comp_type"] == "Slider":
+        new_component = gr.Slider(label=config_dict[component_key]["label"], minimum=config_dict[component_key]["minimum"], maximum=config_dict[component_key]["maximum"], step=config_dict[component_key]["step"])
     else:
         print(
             f"Found component type {config_dict[component_key]['comp_type']} for {component_key}, which is not supported"