Added dropdown for selecting a generator model, commented out length slider.

app.py

@@ -45,7 +45,13 @@ import PIL
 
 
 # ###### Generate ECGs ######################################################
-def predict(numberOfECGs = 1, ecgLengthInSeconds = 10, ecgTypeString = 'ECG-12') -> list:
+def predict(numberOfECGs       = 1,
+            # ecgLengthInSeconds = 10,
+            ecgTypeString      = 'ECG-12',
+            generatorModel     = 'Default',
+           ) -> list:
+
+   ecgLengthInSeconds = 10
 
    # ====== Set ECG type ====================================================
    ecgType = deepfakeecg.DATA_ECG12
@@ -189,10 +195,11 @@ with gradio.Blocks(css = css, theme = gradio.themes.Glass(secondary_hue=gradio.t
 """)
    gradio.Markdown('## Settings')
    with gradio.Row():
-      sliderNumberOfECGs    = gradio.Slider(1, 100, label="Number of ECGs", step = 1, value = 4, interactive = True)
-      sliderLengthInSeconds = gradio.Slider(5, 60, label="Length (s)", step = 5, value = 10, interactive = True)
-      dropdownType          = gradio.Dropdown( [ 'ECG-12', 'ECG-8' ], label = 'ECG Type', interactive = True)
-      buttonGenerate        = gradio.Button("Generate")
+      sliderNumberOfECGs     = gradio.Slider(1, 100, label="Number of ECGs", step = 1, value = 4, interactive = True)
+      # sliderLengthInSeconds = gradio.Slider(5, 60, label="Length (s)", step = 5, value = 10, interactive = True)
+      dropdownType           = gradio.Dropdown( [ 'ECG-12', 'ECG-8' ], label = 'ECG Type', interactive = True)
+      dropdownGeneratorModel = gradio.Dropdown( [ 'Default' ], label = 'Generator Model', interactive = True)
+      buttonGenerate         = gradio.Button("Generate")
    gradio.Markdown('## Output')
    with gradio.Row():
       outputGallery = gradio.Gallery(label = 'output', columns = [ 1 ], height = 'auto',
@@ -201,13 +208,19 @@ with gradio.Blocks(css = css, theme = gradio.themes.Glass(secondary_hue=gradio.t
 
    # ====== Add click event handling for "Generate" button ==================
    buttonGenerate.click(predict,
-                        inputs  = [ sliderNumberOfECGs, sliderLengthInSeconds, dropdownType ],
+                        inputs  = [ sliderNumberOfECGs,
+                                    # sliderLengthInSeconds,
+                                    dropdownType,
+                                    dropdownGeneratorModel ],
                         outputs = [ outputGallery ]
                        )
 
    # ====== Run on startup ==================================================
    gui.load(predict,
-            inputs  = [ sliderNumberOfECGs, sliderLengthInSeconds, dropdownType ],
+            inputs  = [ sliderNumberOfECGs,
+                        # sliderLengthInSeconds,
+                        dropdownType,
+                        dropdownGeneratorModel ],
             outputs = [ outputGallery ]
            )