Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 190 |
for cls in CLASS_LIST:
|
| 191 |
-
|
|
|
|
| 192 |
|
| 193 |
-
|
| 194 |
|
| 195 |
-
|
|
|
|
| 196 |
|
| 197 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|