Update app.py

app.py

@@ -65,7 +65,9 @@ def compile_model(model_name, sram_size, tensor_size, optimize, model_loc, clock
 #        batch = 16
 #        print(f'ERROR converting {epochs_text}, and {batch_text}')
     return output_text
-    
+
+
+model_choices = glob.glob('models/*.tflite')
     
 def update_sliders(sram_slider_value, tensor_slider_value):
 
@@ -86,14 +88,19 @@ with gr.Blocks() as demo:
     optimize = gr.Radio(choices=["Performance", "Size"], value='Performance', label='Performance model')
     model_loc = gr.Radio(choices=["sram", "flash"], value="sram", label='Model weights target')
     
-
     # Setup model compile
-    model_text = gr.Textbox(label='TFLite model', value='hello_world.tflite')
+    model_dropdown = gr.Dropdown(
+        label="Select an model",
+        value='models/hello_world.tflite',
+        choices=model_choices
+    )
+    
+    # Run the compile
     compile_btn = gr.Button("Compile Model")
     compile_text = gr.Markdown("Waiting for model results")
     
     # Compute options
-    compile_btn.click(compile_model, inputs=[model_text, sram_slider, tensor_slider, optimize, model_loc, clock_slider], outputs=[compile_text])
+    compile_btn.click(compile_model, inputs=[model_dropdown, sram_slider, tensor_slider, optimize, model_loc, clock_slider], outputs=[compile_text])
     sram_slider.change(fn=update_sliders, inputs=[sram_slider, tensor_slider], outputs=tensor_slider)
     demo.load(get_oauth_info, inputs=None, outputs=user_text)