IlyasMoutawwakil HF staff commited on
Commit
b430e89
1 Parent(s): 169283f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,14 +72,14 @@ def parse_configs(inputs):
72
  task=task,
73
  model=model,
74
  device=DEVICE,
75
- **{k: v for k, v in configs["pytorch"] if v},
76
  )
77
 
78
  configs["openvino"] = OVConfig(
79
  task=task,
80
  model=openvino_label or model,
81
  device=DEVICE,
82
- **{k: v for k, v in configs["openvino"] if v},
83
  )
84
 
85
  return configs
 
72
  task=task,
73
  model=model,
74
  device=DEVICE,
75
+ **{k: v for k, v in configs["pytorch"].items() if v},
76
  )
77
 
78
  configs["openvino"] = OVConfig(
79
  task=task,
80
  model=openvino_label or model,
81
  device=DEVICE,
82
+ **{k: v for k, v in configs["openvino"].items() if v},
83
  )
84
 
85
  return configs