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 * 1024)
-        lpmem_size_limit = int(lpmem_value * 1024)
+        vmem_size_limit = int(vmem_value * 1024000)
+        lpmem_size_limit = int(lpmem_value * 1024000)
         
         # Run the model fitter
         success, results = sr100_model_compiler.sr100_model_optimizer(
@@ -48,28 +48,14 @@ def compile_model(model_name, vmem_value, lpmem_value):
         )
         print(results)
 
-    # Analyze the model
-    #default_config = sr100_model_compiler.sr100_default_config()
-
-    #default_config['sram_size'] = int(float(sram_size)*1.0e6)
-    #default_config['core_clock'] = int(float(clock)*1.0e6)
-    #success, perf_data = sr100_model_compiler.sr100_check_model(results=results, config=default_config)
-
     output_text = ''
     if success:
         output_text = 'SUCCESS, model fits on SR100'
     else:
         output_text = 'FAILULRE model does not fit on SR100'
     for key, value in results.items():
-        output_text += f'<br>{key} = {value}'
+        output_text += f'<br>    {key} = {value}'
     
-#    try:
-#        epochs = int(epochs_text)
-#        batch = int(batch_text)
-#    except Exception as e:
-#        epochs = 100
-#        batch = 16
-#        print(f'ERROR converting {epochs_text}, and {batch_text}')
     return output_text
 
 # Get all available models