lewtun HF staff commited on
Commit
b6fc567
1 Parent(s): 449b6ea

Fix AutoTrain provider

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -576,7 +576,7 @@ with st.form(key="form"):
576
  else "en",
577
  "max_models": 5,
578
  "instance": {
579
- "provider": "sagemaker",
580
  "instance_type": AUTOTRAIN_MACHINE[selected_task]
581
  if selected_task in AUTOTRAIN_MACHINE.keys()
582
  else "p3",
 
576
  else "en",
577
  "max_models": 5,
578
  "instance": {
579
+ "provider": "sagemaker" if selected_task in AUTOTRAIN_MACHINE.keys() else "ovh",
580
  "instance_type": AUTOTRAIN_MACHINE[selected_task]
581
  if selected_task in AUTOTRAIN_MACHINE.keys()
582
  else "p3",