fix a bug

app.py

@@ -369,6 +369,7 @@ with demo:
                     private,
                     weight_type,
                     model_type,
+                    inference_framework,
                 ],
                 submission_result,
             )

src/leaderboard/read_evals.py

@@ -49,8 +49,6 @@ class EvalResult:
         with open(json_filepath) as fp:
             data = json.load(fp)
 
-        inference_framework = data.get("inference_framework", "Unknown")
-
         # We manage the legacy config format
         config = data.get("config", data.get("config_general", None))
 
@@ -61,6 +59,9 @@ class EvalResult:
         org_and_model = config.get("model_name", config.get("model_args", None))
         org_and_model = org_and_model.split("/", 1)
 
+        # Get inference framework
+        inference_framework = config.get("inference_framework", "Unknown")
+
         if len(org_and_model) == 1:
             org = None
             model = org_and_model[0]

src/submission/submit.py

@@ -25,6 +25,7 @@ def add_new_eval(
     private: bool,
     weight_type: str,
     model_type: str,
+    inference_framework: str,
 ):
     global REQUESTED_MODELS
     global USERS_TO_SUBMISSION_DATES
@@ -85,6 +86,8 @@ def add_new_eval(
         license = model_info.cardData["license"]
     except Exception:
         return styled_error("Please select a license for your model")
+    
+    # TODO: Check if the inference framework is valid
 
     modelcard_OK, error_msg = check_model_card(model)
     if not modelcard_OK:
@@ -106,6 +109,7 @@ def add_new_eval(
         "likes": model_info.likes,
         "params": model_size,
         "license": license,
+        "inference_framework": inference_framework,
     }
 
     # Check for duplicate submission