chansung commited on
Commit
31bb77c
·
1 Parent(s): fd2a227

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -15
app.py CHANGED
@@ -182,15 +182,43 @@ 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": {
@@ -198,18 +226,18 @@ def submit(
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)
214
 
215
  payload = json.dumps(payload)
 
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
+ {
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
  "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)
242
 
243
  payload = json.dumps(payload)