change inf type to k-v pair

app.py

@@ -266,7 +266,7 @@ with gr.Blocks(theme=theme) as iface:
                 ''')
         with gr.Row():
             run_local = gr.Checkbox(value=True, label="Run in this Space")
-            use_inference = read_inference_type('./config.yaml')[0] == 'hf_inference_api'
+            use_inference = read_inference_type('./config.yaml') == 'hf_inference_api'
             run_inference = gr.Checkbox(value=use_inference, label="Run with Inference API")
         
         with gr.Row() as advanced_row:

config.yaml

@@ -6,5 +6,4 @@ detectors:
   - underconfidence
   - overconfidence
   - spurious_correlation
-inference_type:
-  - hf_inference_api
+inference_type: hf_pipeline

utils.py

@@ -38,9 +38,9 @@ def write_inference_type(use_inference):
     with open(YAML_PATH, "r") as f:
         config = yaml.load(f, Loader=yaml.FullLoader)
     if use_inference:
-        config["inference_type"] = ['hf_inference_api']
+        config["inference_type"] = 'hf_inference_api'
     else:
-        config["inference_type"] = ['hf_pipeline']
+        config["inference_type"] = 'hf_pipeline'
     with open(YAML_PATH, "w") as f:
         # save inference_type to inference_type in yaml
         yaml.dump(config, f, Dumper=Dumper)