chansung commited on
Commit
32e7a76
1 Parent(s): b7fa83a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -18
app.py CHANGED
@@ -127,20 +127,6 @@ def update_regions(provider):
127
  )
128
 
129
  def update_compute_options(provider, regeion):
130
- # CPU [medium] 路 2vCPU 4GB 路 Intel Ice Lake
131
- # 'https://api.endpoints.huggingface.cloud/provider/aws/region/us-east-1/compute'
132
- """
133
- "accelerator": "cpu",
134
- "numAccelerators": 1,
135
- "memoryGb": "2Gi",
136
- "instanceType": "c6i",
137
- "instanceSize": "small",
138
- "architecture": "Intel Ice Lake",
139
- "status": "available",
140
- "pricePerHour": 0.06
141
-
142
- """
143
-
144
  avalialbe_compute_options = []
145
 
146
  headers = {
@@ -151,10 +137,19 @@ def update_compute_options(provider, regeion):
151
 
152
  for compute in response.json()['items']:
153
  if compute['status'] == 'avaliable':
 
 
 
 
154
  avalialbe_compute_options.append(
155
- f"{compute['accelerator'].upper()} []"
156
  )
157
 
 
 
 
 
 
158
 
159
  with gr.Blocks() as demo2:
160
  gr.Markdown(
@@ -169,9 +164,9 @@ with gr.Blocks() as demo2:
169
 
170
  with gr.Row():
171
  provider_selector = gr.Dropdown(
 
172
  label="select cloud provider",
173
  interactive=True,
174
- choices=providers
175
  )
176
 
177
  region_selector = gr.Dropdown(
@@ -180,9 +175,16 @@ with gr.Blocks() as demo2:
180
  interactive=True,
181
  label="select a region"
182
  )
183
- provider_selector.change(update_regions, inputs=provider_selector, outputs=region_selector)
184
- # region_selector.update(interactive=True)
185
 
 
 
 
 
 
 
 
 
 
186
 
187
  gr.TabbedInterface(
188
  [demoInterface, demo, demo2], ["Try-out", "馃殌 Deploy on GCP", " Deploy on 馃 Endpoint"]
 
127
  )
128
 
129
  def update_compute_options(provider, regeion):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
130
  avalialbe_compute_options = []
131
 
132
  headers = {
 
137
 
138
  for compute in response.json()['items']:
139
  if compute['status'] == 'avaliable':
140
+ type = f"{compute['numAccelerators']}vCPU {compute['memoryGb'].replace("Gi", "GB")} 路 {compute['architecture']}"
141
+ if compute['accelerator'] == "cpu"
142
+ else f"{compute['numAccelerators']x {compute['architecture']}}"
143
+
144
  avalialbe_compute_options.append(
145
+ f"{compute['accelerator'].upper()} [{compute['instanceSize']}] 路 {type}"
146
  )
147
 
148
+ return gr.Dropdown.update(
149
+ choices=avalialbe_compute_options,
150
+ value=avalialbe_compute_options[0] if len(avalialbe_compute_options) > 0 else None
151
+ )
152
+
153
 
154
  with gr.Blocks() as demo2:
155
  gr.Markdown(
 
164
 
165
  with gr.Row():
166
  provider_selector = gr.Dropdown(
167
+ providers
168
  label="select cloud provider",
169
  interactive=True,
 
170
  )
171
 
172
  region_selector = gr.Dropdown(
 
175
  interactive=True,
176
  label="select a region"
177
  )
 
 
178
 
179
+ provider_selector.change(update_regions, inputs=provider_selector, outputs=region_selector)
180
+
181
+ compute_selector = gr.Dropdown(
182
+ [],
183
+ value="",
184
+ label="select compute instance type"
185
+ interactive=True
186
+ )
187
+ region_selector.change(update_compute_options, inputs=[provider_selector, region_selector], outputs=compute_selector)
188
 
189
  gr.TabbedInterface(
190
  [demoInterface, demo, demo2], ["Try-out", "馃殌 Deploy on GCP", " Deploy on 馃 Endpoint"]