Update app.py

app.py

@@ -363,6 +363,12 @@ with demo:
                     label="Base Model Name",
                     interactive = True,
                 )  
+                PEFT_model_name = gr.Dropdown(
+                    choices=PEFT_MODEL_OPTIONS,
+                    value=PEFT_MODEL_OPTIONS[0],
+                    label="PEFT Model Name",
+                    interactive = True,
+                )  
             with gr.Column(scale=0.5, variant="compact"):    
                 finetune_button = gr.Button(
                     value="Finetune Pre-trained Model",