badaoui HF Staff commited on
Commit
e9c6162
·
verified ·
1 Parent(s): a03bf63

tag diffusers as (soon)

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -190,7 +190,7 @@ def neuron_export(model_id: str, model_type: str, pipeline_name: str, task_or_pi
190
  for status_code, message in convert(
191
  api, model_id, task_or_pipeline, model_type,
192
  token=oauth_token.token, pr_options=pr_options,
193
- pipeline_name=pipeline_name if model_type == "diffusers" else None
194
  ):
195
  if isinstance(message, str):
196
  yield log(message)
@@ -319,7 +319,7 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft()) as demo:
319
  )
320
  with gr.Row():
321
  model_type = gr.Radio(
322
- choices=["transformers", "diffusers"],
323
  value="transformers",
324
  label="Model Type",
325
  info="Choose the type of model you want to export"
 
190
  for status_code, message in convert(
191
  api, model_id, task_or_pipeline, model_type,
192
  token=oauth_token.token, pr_options=pr_options,
193
+ pipeline_name=pipeline_name if model_type == "diffusers (soon)" else None
194
  ):
195
  if isinstance(message, str):
196
  yield log(message)
 
319
  )
320
  with gr.Row():
321
  model_type = gr.Radio(
322
+ choices=["transformers", "diffusers (soon)"],
323
  value="transformers",
324
  label="Model Type",
325
  info="Choose the type of model you want to export"