Spaces:
Running
Running
tag diffusers as (soon)
Browse files
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"
|