Update app.py

app.py

@@ -11,6 +11,7 @@ get_ranking_url = f"{leaderboard_url}/model/get-ranking-by-category"
 get_models_url = f"{leaderboard_url}/model/get-models-by-category"
 vote_url = f"{leaderboard_url}/model/vote"
 submit_models_url = f"{leaderboard_url}/model/submit-models"
+submit_category_url = f"{leaderboard_url}/model/submit-category"
 
 def make_clickable_model(model_id):
     model_name_show = ' '.join(model_id.split('/')[1:])
@@ -90,17 +91,24 @@ def submit_model(category, customize_category, model_id):
     if not model_id:
         return "All fields are required!"
     if not category or not customize_category:
-        return "Please choose a cateogry or customize your own category!"
-    
-    selected_category = customize_category if customize_category else category
+        return "Please choose a category or customize your own category!"
+
+    selected_category = category
+
+    if customize_category:
+        category_data = {
+            "category": customize_category
+        }
+        response = requests.post(submit_category_url, json=category_data)
+        if response.status_code != 200:
+            return f"Failed to submit category: {response.text}"
+        selected_category = customize_category
 
     data = {
         "model_id": model_id,
         "categories": [selected_category]
     }
-    
     response = requests.post(submit_models_url, json=data)
-    
     if response.status_code == 200:
         return "Your request has been submitted successfully. We will notify you by email once processing is complete."
     else: