Update app.py

app.py

@@ -37,8 +37,8 @@ def compile_model(model_name, vmem_value, lpmem_value):
     with tempfile.TemporaryDirectory() as out_dir:
         print(f"Created temporary directory: {out_dir}")
 
-        vmem_size_limit = int(vmem_value * 1024000)
-        lpmem_size_limit = int(lpmem_value * 1024000)
+        vmem_size_limit = int(vmem_value * 1000)
+        lpmem_size_limit = int(lpmem_value * 1000)
         
         # Run the model fitter
         success, results = sr100_model_compiler.sr100_model_optimizer(
@@ -60,24 +60,24 @@ def compile_model(model_name, vmem_value, lpmem_value):
         else:
             output_text = 'FAILURE, model does not fit on SR100'
 
-        weights_size = results['weights_size'] / 1024.0
-        arena_size = results['arena_cache_size'] / 1024.0
+        weights_size = results['weights_size'] / 1000.0
+        arena_size = results['arena_cache_size'] / 100.0
         clock = results['core_clock']  / 1.0e6
         
         output_text += f'<br>clock = {clock:0.1f} MHz'
-        output_text += f'<br>Model size = {weights_size:0.1f} KiB'
-        output_text += f'<br>Arena cache size = {arena_size:0.1f} KiB'
-        output_text += f'<br>model loc {results["model_loc"]}'
-        #output_text += f'<br>System config {results["system_config"]}'
+        output_text += f'<br>Model size = {weights_size:0.3f} kB'
+        output_text += f'<br>Arena cache size = {arena_size:0.3f} kB'
+        output_text += f'<br>model loc = {results["model_loc"]}'
+        output_text += f'<br>System config = {results["system_config"]}'
 
 
-        vmem_size = results['vmem_size'] / 1024.0
-        lpmem_size = results['lpmem_size'] / 1024.0
+        vmem_size = results['vmem_size'] / 1000.0
+        lpmem_size = results['lpmem_size'] / 1000.0
         vmem_perc = results['vmem_size'] * 100.0 / results['vmem_size_limit']
         lpmem_perc = results['lpmem_size'] * 100.0 / results['lpmem_size_limit']
         
-        output_text += f'<br>vmem_size = {vmem_size:0.1f} KiB ({vmem_perc:0.1f}%)'
-        output_text += f'<br>lpmem_size = {lpmem_size:0.1f} KiB ({lpmem_perc:0.1f}%)'
+        output_text += f'<br>vmem_size = {vmem_size:0.3f} kB ({vmem_perc:0.1f}%)'
+        output_text += f'<br>lpmem_size = {lpmem_size:0.3f} kB ({lpmem_perc:0.1f}%)'
 
     
     return output_text
@@ -92,8 +92,8 @@ with gr.Blocks() as demo:
 
     # Setup model inputs
     with gr.Row():
-        vmem_slider = gr.Slider(minimum=0, maximum=1.5, step=0.1, label="Set total VMEM SRAM size available in MiB", value=1.5)
-        lpmem_slider = gr.Slider(minimum=0, maximum=1.5, step=0.1, label="Set total LPMEM SRAM size in MiB", value=1.5)
+        vmem_slider = gr.Slider(minimum=0, maximum=1536, step=1.024, label="Set total VMEM SRAM size available in kB", value=1536.0)
+        lpmem_slider = gr.Slider(minimum=0, maximum=1536, step=1.024, label="Set total LPMEM SRAM size in kB", value=1536.0)
     
     # Setup model compile
     model_dropdown = gr.Dropdown(