aibota01 commited on
Commit
c7730e8
·
1 Parent(s): 9174245
Files changed (2) hide show
  1. app.py +4 -6
  2. requirements.txt +1 -1
app.py CHANGED
@@ -174,12 +174,10 @@ def evaluate_submission(file, submission_name, model_description):
174
  return f"❌ Error during merge for MAE: {e}"
175
 
176
  # Convert weight columns to numeric.
177
- df["weight_gt"] = pd.to_numeric(df["weight"], errors="coerce")
178
- df["weight_pred"] = pd.to_numeric(df["weight_pred"], errors="coerce") # Note: prediction file column might be named "weight" as well.
179
- # If prediction CSV uses the same column name 'weight', adjust accordingly:
180
- if "weight_pred" not in df.columns:
181
- df.rename(columns={"weight": "weight_pred"}, inplace=True)
182
- df["weight_gt"] = pd.to_numeric(gt_df["weight"], errors="coerce")
183
 
184
  # Filter out rows where either weight is -1.
185
  before_filter = df.shape[0]
 
174
  return f"❌ Error during merge for MAE: {e}"
175
 
176
  # Convert weight columns to numeric.
177
+ # After merging, the ground truth weight is in "weight_gt" and prediction weight in "weight_pred"
178
+ df["weight_gt"] = pd.to_numeric(df["weight_gt"], errors="coerce")
179
+ df["weight_pred"] = pd.to_numeric(df["weight_pred"], errors="coerce")
180
+
 
 
181
 
182
  # Filter out rows where either weight is -1.
183
  before_filter = df.shape[0]
requirements.txt CHANGED
@@ -5,4 +5,4 @@ APScheduler==3.9.1
5
  huggingface-hub>=0.10.0
6
  pydantic==2.10.6
7
  scikit-learn==1.3.2
8
- pycocotools
 
5
  huggingface-hub>=0.10.0
6
  pydantic==2.10.6
7
  scikit-learn==1.3.2
8
+ pycocotools==2.0.8