Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,14 +113,14 @@ def convert_checkpoint_to_safetensors(checkpoint_url: str, model_name: str):
|
|
| 113 |
|
| 114 |
title = "Checkpoint to SafeTensors Converter"
|
| 115 |
description = """
|
| 116 |
-
Convert your PyTorch Lightning
|
| 117 |
|
| 118 |
**Inputs**:
|
| 119 |
-
- **Checkpoint URL**: Direct link to the
|
| 120 |
-
- **Model Name**: Name of the base model (e.g.,
|
| 121 |
|
| 122 |
**Output**:
|
| 123 |
-
- Downloadable
|
| 124 |
"""
|
| 125 |
|
| 126 |
iface = gr.Interface(
|
|
@@ -143,16 +143,6 @@ iface = gr.Interface(
|
|
| 143 |
],
|
| 144 |
title=title,
|
| 145 |
description=description,
|
| 146 |
-
examples=[
|
| 147 |
-
[
|
| 148 |
-
"https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-base.ckpt?download=true",
|
| 149 |
-
"roberta-base"
|
| 150 |
-
],
|
| 151 |
-
[
|
| 152 |
-
"https://path.to/your/checkpoint.ckpt",
|
| 153 |
-
"bert-base-uncased"
|
| 154 |
-
]
|
| 155 |
-
],
|
| 156 |
allow_flagging="never"
|
| 157 |
)
|
| 158 |
|
|
|
|
| 113 |
|
| 114 |
title = "Checkpoint to SafeTensors Converter"
|
| 115 |
description = """
|
| 116 |
+
Convert your PyTorch Lightning .ckpt checkpoints to the secure safetensors format.
|
| 117 |
|
| 118 |
**Inputs**:
|
| 119 |
+
- **Checkpoint URL**: Direct link to the .ckpt file.
|
| 120 |
+
- **Model Name**: Name of the base model (e.g., roberta-base, bert-base-uncased).
|
| 121 |
|
| 122 |
**Output**:
|
| 123 |
+
- Downloadable safetensors file.
|
| 124 |
"""
|
| 125 |
|
| 126 |
iface = gr.Interface(
|
|
|
|
| 143 |
],
|
| 144 |
title=title,
|
| 145 |
description=description,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 146 |
allow_flagging="never"
|
| 147 |
)
|
| 148 |
|