chansung commited on
Commit
29b6f8f
·
1 Parent(s): bbb7e21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -42
app.py CHANGED
@@ -182,43 +182,15 @@ def submit(
182
 
183
  type = compute_resources[-1].strip()
184
 
185
- # payload = {
186
- # "accountId": repository_selector.split("/")[0],
187
- # "compute": {
188
- # "accelerator": accelerator,
189
- # "instanceSize": size[1:],
190
- # "instanceType": type,
191
- # "scaling": {
192
- # "maxReplica": int(max_node_selector),
193
- # "minReplica": int(min_node_selector)
194
- # }
195
- # },
196
- # "model": {
197
- # "framework": "custom",
198
- # "image": {
199
- # "huggingface": {}
200
- # },
201
- # "repository": repository_selector.split("/")[1].lower(),
202
- # "revision": head_sha,
203
- # "task": task_selector.lower()
204
- # },
205
- # "name": endpoint_name_input.strip(),
206
- # "provider": {
207
- # "region": region_selector.split("/")[0].lower(),
208
- # "vendor": provider_selector.lower()
209
- # },
210
- # "type": security_selector.lower()
211
- # }
212
-
213
  payload = {
214
- "accountId": "chansung",
215
  "compute": {
216
- "accelerator": "cpu",
217
- "instanceSize": "large",
218
- "instanceType": "c6i",
219
  "scaling": {
220
- "maxReplica": 8,
221
- "minReplica": 2
222
  }
223
  },
224
  "model": {
@@ -226,16 +198,16 @@ def submit(
226
  "image": {
227
  "huggingface": {}
228
  },
229
- "repository": "gpt2",
230
- "revision": "6c0e6080953db56375760c0471a8c5f2929baf11",
231
- "task": "text-classification"
232
  },
233
- "name": "my-endpoint",
234
  "provider": {
235
- "region": "us-east-1",
236
- "vendor": "aws"
237
  },
238
- "type": "public"
239
  }
240
 
241
  print(payload)
@@ -273,7 +245,8 @@ with gr.Blocks() as demo2:
273
  #### Your 🤗 Access Token
274
  """)
275
  hf_token_input = gr.Textbox(
276
- show_label=False
 
277
  )
278
 
279
  gr.Markdown("""
 
182
 
183
  type = compute_resources[-1].strip()
184
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
185
  payload = {
186
+ "accountId": repository_selector.split("/")[0],
187
  "compute": {
188
+ "accelerator": accelerator,
189
+ "instanceSize": size[1:],
190
+ "instanceType": type,
191
  "scaling": {
192
+ "maxReplica": int(max_node_selector),
193
+ "minReplica": int(min_node_selector)
194
  }
195
  },
196
  "model": {
 
198
  "image": {
199
  "huggingface": {}
200
  },
201
+ "repository": repository_selector.split("/")[1].lower(),
202
+ "revision": head_sha,
203
+ "task": task_selector.lower()
204
  },
205
+ "name": endpoint_name_input.strip(),
206
  "provider": {
207
+ "region": region_selector.split("/")[0].lower(),
208
+ "vendor": provider_selector.lower()
209
  },
210
+ "type": security_selector.lower()
211
  }
212
 
213
  print(payload)
 
245
  #### Your 🤗 Access Token
246
  """)
247
  hf_token_input = gr.Textbox(
248
+ show_label=False,
249
+ type="password"
250
  )
251
 
252
  gr.Markdown("""