Spaces:

raman07
/

MIMIC-SD-Demo-Memory-Optimized

Sleeping

Raman Dutt commited on Jan 31, 2024

Commit

d530147

•

1 Parent(s): 58122ec

minor fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ def predict(
     # Create a Bar Plot displaying the number of tunable parameters for the selected PEFT Type
     df = pd.DataFrame(
         {
-            "PEFT Type": list(NUM_TUNABLE_PARAMS.keys()),
             "Number of Tunable Parameters": list(NUM_TUNABLE_PARAMS.values()),
         }
     )

     # Create a Bar Plot displaying the number of tunable parameters for the selected PEFT Type
     df = pd.DataFrame(
         {
+            "Fine-Tuning Strategy": list(NUM_TUNABLE_PARAMS.keys()),
             "Number of Tunable Parameters": list(NUM_TUNABLE_PARAMS.values()),
         }
     )