Change the name of output model to `<model_name>-ONNX`
Browse filesPreviously it was `<author_name>-<model_name>-ONNX`.
app.py
CHANGED
|
@@ -171,13 +171,7 @@ def main():
|
|
| 171 |
key="user_hf_token",
|
| 172 |
)
|
| 173 |
|
| 174 |
-
model_name = (
|
| 175 |
-
input_model_id.replace(f"{config.hf_base_url}/", "")
|
| 176 |
-
.replace("/", "-")
|
| 177 |
-
.replace(f"{config.hf_username}-", "")
|
| 178 |
-
.strip()
|
| 179 |
-
)
|
| 180 |
-
|
| 181 |
output_model_id = f"{config.hf_username}/{model_name}-ONNX"
|
| 182 |
output_model_url = f"{config.hf_base_url}/{output_model_id}"
|
| 183 |
|
|
|
|
| 171 |
key="user_hf_token",
|
| 172 |
)
|
| 173 |
|
| 174 |
+
model_name = input_model_id.split("/")[-1]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 175 |
output_model_id = f"{config.hf_username}/{model_name}-ONNX"
|
| 176 |
output_model_url = f"{config.hf_base_url}/{output_model_id}"
|
| 177 |
|