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 |
|