bar plot params changed

app.py

@@ -107,6 +107,7 @@ def predict(
     OUTPUT_DIR="OUTPUT",
 ):
 
+    BARPLOT_TITLE = "Tunable Parameters for {} Fine-Tuning".format(unet_pretraining_type)
     NUM_TUNABLE_PARAMS = {
         "full": 86,
         "attention": 26.7,
@@ -139,8 +140,6 @@ def predict(
     result_pil_image = result_image["images"][0]
 
     # Create a Bar Plot displaying the number of tunable parameters for the selected PEFT Type
-    # Create a Pandas DataFrame
-    
     df = pd.DataFrame(
         {
             "PEFT Type": list(NUM_TUNABLE_PARAMS.keys()),
@@ -154,13 +153,14 @@ def predict(
 
     bar_plot = gr.BarPlot(
         value=df,
-        x="PEFT Type",
+        x="Fine-Tuning Strategy",
         y="Number of Tunable Parameters",
-        label="PEFT Type",
         title="Number of Tunable Parameters",
         vertical=True,
         height=300,
         width=300,
+        title=BARPLOT_TITLE,
+        interactive=True,
     )
 
     return result_pil_image, bar_plot