Felladrin commited on
Commit
811a3ef
·
1 Parent(s): 0bdd7b0

Change the name of output model to `<model_name>-ONNX`

Browse files

Previously it was `<author_name>-<model_name>-ONNX`.

Files changed (1) hide show
  1. app.py +1 -7
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