JunJiaGuo commited on
Commit
9da963f
·
verified ·
1 Parent(s): 5118e05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -186,15 +186,25 @@ def evaluate_uploaded_json(user_file, model_name):
186
 
187
  def save_class_accuracy_to_hf_dataset(model_name, class_accuracy):
188
 
189
- data = {"Model Name": model_name}
190
  for cls in CLASS_LIST:
191
- data[cls] = class_accuracy.get(cls, 0)
 
192
 
193
- df = pd.DataFrame([data])
194
 
195
- dataset = Dataset.from_pandas(df)
 
196
 
197
- dataset.push_to_hub(HF_DATASET_REPO, split="train")
 
 
 
 
 
 
 
 
198
 
199
 
200
 
 
186
 
187
  def save_class_accuracy_to_hf_dataset(model_name, class_accuracy):
188
 
189
+ new_data = {"Model Name": model_name}
190
  for cls in CLASS_LIST:
191
+ new_data[cls] = class_accuracy.get(cls, 0)
192
+ new_df = pd.DataFrame([new_data])
193
 
194
+ try:
195
 
196
+ dataset = load_dataset(HF_DATASET_REPO, split="train")
197
+ existing_df = dataset.to_pandas()
198
 
199
+
200
+ updated_df = pd.concat([existing_df, new_df], ignore_index=True)
201
+ except:
202
+
203
+ updated_df = new_df
204
+
205
+
206
+ updated_dataset = Dataset.from_pandas(updated_df)
207
+ updated_dataset.push_to_hub(HF_DATASET_REPO, split="train", token=HF_TOKEN)
208
 
209
 
210