lixuejing commited on
Commit
4cc735e
·
1 Parent(s): d7938a7

update model infer

Browse files
Files changed (2) hide show
  1. app.py +5 -3
  2. src/submission/submit.py +2 -0
app.py CHANGED
@@ -420,8 +420,8 @@ with demo:
420
  with gr.Column():
421
  model_name_textbox = gr.Textbox(label="Model name")
422
  model_url_textbox = gr.Textbox(label="Model online api url")
423
- revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
424
- private = gr.Checkbox(False, label="Private", visible=not IS_PUBLIC)
425
  model_type = gr.Dropdown(
426
  choices=[t.to_str(" : ") for t in ModelType if t != ModelType.Unknown],
427
  label="Model type",
@@ -431,7 +431,7 @@ with demo:
431
  )
432
 
433
  with gr.Column():
434
- model_api_key = gr.Textbox(label="Model online api key")
435
  precision = gr.Dropdown(
436
  choices=[i.value.name for i in Precision if i != Precision.Unknown],
437
  label="Precision",
@@ -446,6 +446,7 @@ with demo:
446
  value="Original",
447
  interactive=True,
448
  )
 
449
  base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
450
 
451
  submit_button = gr.Button("Submit Eval")
@@ -456,6 +457,7 @@ with demo:
456
  model_name_textbox,
457
  model_url_textbox,
458
  model_api_key,
 
459
  base_model_name_textbox,
460
  revision_name_textbox,
461
  precision,
 
420
  with gr.Column():
421
  model_name_textbox = gr.Textbox(label="Model name")
422
  model_url_textbox = gr.Textbox(label="Model online api url")
423
+ model_api_key = gr.Textbox(label="Model online api key")
424
+ model_api_name_textbox = gr.Textbox(label="Online api model name")
425
  model_type = gr.Dropdown(
426
  choices=[t.to_str(" : ") for t in ModelType if t != ModelType.Unknown],
427
  label="Model type",
 
431
  )
432
 
433
  with gr.Column():
434
+ private = gr.Checkbox(False, label="Private", visible=not IS_PUBLIC)
435
  precision = gr.Dropdown(
436
  choices=[i.value.name for i in Precision if i != Precision.Unknown],
437
  label="Precision",
 
446
  value="Original",
447
  interactive=True,
448
  )
449
+ revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
450
  base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
451
 
452
  submit_button = gr.Button("Submit Eval")
 
457
  model_name_textbox,
458
  model_url_textbox,
459
  model_api_key,
460
+ model_api_name_textbox,
461
  base_model_name_textbox,
462
  revision_name_textbox,
463
  precision,
src/submission/submit.py CHANGED
@@ -19,6 +19,7 @@ def add_new_eval(
19
  model: str,
20
  model_api_url: str,
21
  model_api_key: str,
 
22
  base_model: str,
23
  revision: str,
24
  precision: str,
@@ -107,6 +108,7 @@ def add_new_eval(
107
  "model": model,
108
  "model_api_url": model_api_url,
109
  "model_api_key": model_api_key,
 
110
  "base_model": base_model,
111
  "revision": revision,
112
  "precision": precision,
 
19
  model: str,
20
  model_api_url: str,
21
  model_api_key: str,
22
+ model_api_name: str,
23
  base_model: str,
24
  revision: str,
25
  precision: str,
 
108
  "model": model,
109
  "model_api_url": model_api_url,
110
  "model_api_key": model_api_key,
111
+ "model_api_name": model_api_name,
112
  "base_model": base_model,
113
  "revision": revision,
114
  "precision": precision,