Spaces:

Nidhal-ch
/

test-pfe

Sleeping

Nidhal-ch commited on Jun 9, 2024

Commit

c3200ea

verified ·

1 Parent(s): 36ea705

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -129,10 +129,10 @@ with col1:
     if model in ['SSD300', 'SSD512']:
         df_results = pd.DataFrame({
             'Model': results[pruning]['model'],
-            'mAP': results[pruning]['map'],
-            'FLOPs': results[pruning]['flops'],
             'down by': results[pruning]['flopsd'],
-            'Params': results[pruning]['params'],
             'down by ': results[pruning]['paramsd'],
         })
     else:

     if model in ['SSD300', 'SSD512']:
         df_results = pd.DataFrame({
             'Model': results[pruning]['model'],
+            'mAP (%)': results[pruning]['map'],
+            'GFLOPs': results[pruning]['flops'],
             'down by': results[pruning]['flopsd'],
+            'Parameters (M)': results[pruning]['params'],
             'down by ': results[pruning]['paramsd'],
         })
     else: