Spaces:
Runtime error
Runtime error
Fix AutoTrain provider
Browse files
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",
|