Upload to same repo (#14)

- Make it possible to upload to same repo (c09521cccb0f1a24ea8203eb079e829a00eccbc2)


Co-authored-by: Pierre Mesure <[email protected]>

app.py

@@ -158,11 +158,23 @@ def main():
             key="user_hf_token",
         )
 
+        if config.hf_username == input_model_id.split("/")[0]:
+            same_repo = st.checkbox(
+                "Do you want to upload the ONNX weights to the same repository?"
+            )
+        else:
+            same_repo = False
+
         model_name = input_model_id.split("/")[-1]
-        output_model_id = f"{config.hf_username}/{model_name}-ONNX"
+
+        output_model_id = f"{config.hf_username}/{model_name}"
+
+        if not same_repo:
+            output_model_id += "-ONNX"
+
         output_model_url = f"{config.hf_base_url}/{output_model_id}"
 
-        if converter.api.repo_exists(output_model_id):
+        if not same_repo and converter.api.repo_exists(output_model_id):
             st.write("This model has already been converted! 🎉")
             st.link_button(f"Go to {output_model_id}", output_model_url, type="primary")
             return