Sreerama commited on
Commit
638a08d
1 Parent(s): a00d107

Fix input index

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -211,9 +211,9 @@ def train(*inputs):
211
  gr.update(visible=True) #completed_training
212
  ]
213
  else:
214
- hf_token = inputs[-5]
215
- model_name = inputs[-7]
216
- where_to_upload = inputs[-8]
217
  push(model_name, where_to_upload, hf_token, which_model, True)
218
  hardware_url = f"https://huggingface.co/spaces/{os.environ['SPACE_ID']}/hardware"
219
  headers = { "authorization" : f"Bearer {hf_token}"}
211
  gr.update(visible=True) #completed_training
212
  ]
213
  else:
214
+ hf_token = inputs[-4]
215
+ model_name = inputs[-6]
216
+ where_to_upload = inputs[-7]
217
  push(model_name, where_to_upload, hf_token, which_model, True)
218
  hardware_url = f"https://huggingface.co/spaces/{os.environ['SPACE_ID']}/hardware"
219
  headers = { "authorization" : f"Bearer {hf_token}"}