Spaces:

autoevaluate
/

model-evaluator

Running

lewtun HF staff commited on Sep 13, 2022

Commit

b6fc567

•

1 Parent(s): 449b6ea

Fix AutoTrain provider

Files changed (1) hide show

app.py CHANGED Viewed

@@ -576,7 +576,7 @@ with st.form(key="form"):
                         else "en",
                         "max_models": 5,
                         "instance": {
-                            "provider": "sagemaker",
                             "instance_type": AUTOTRAIN_MACHINE[selected_task]
                             if selected_task in AUTOTRAIN_MACHINE.keys()
                             else "p3",

                         else "en",
                         "max_models": 5,
                         "instance": {
+                            "provider": "sagemaker" if selected_task in AUTOTRAIN_MACHINE.keys() else "ovh",
                             "instance_type": AUTOTRAIN_MACHINE[selected_task]
                             if selected_task in AUTOTRAIN_MACHINE.keys()
                             else "p3",