Update app.py
Browse files
app.py
CHANGED
|
@@ -80,8 +80,8 @@ def compile_model(model_name, vmem_value, lpmem_value):
|
|
| 80 |
vmem_perc = results['vmem_size'] * 100.0 / results['vmem_size_limit']
|
| 81 |
lpmem_perc = results['lpmem_size'] * 100.0 / results['lpmem_size_limit']
|
| 82 |
|
| 83 |
-
output_text += f'<br>vmem_size = {vmem_size:0.3f} kB ({vmem_perc:0.1f}% of {vmem_size_limit:0.3f} kB)'
|
| 84 |
-
output_text += f'<br>lpmem_size = {lpmem_size:0.3f} kB ({lpmem_perc:0.1f}% of {lpmem_size_limit:0.3f} kB)'
|
| 85 |
|
| 86 |
|
| 87 |
return output_text
|
|
@@ -91,7 +91,7 @@ model_choices = glob.glob('models/*.tflite')
|
|
| 91 |
|
| 92 |
with gr.Blocks() as demo:
|
| 93 |
#gr.LoginButton()
|
| 94 |
-
text1 = gr.Markdown("SR100 Model
|
| 95 |
#user_text = gr.Markdown("")
|
| 96 |
|
| 97 |
# Setup model inputs
|
|
|
|
| 80 |
vmem_perc = results['vmem_size'] * 100.0 / results['vmem_size_limit']
|
| 81 |
lpmem_perc = results['lpmem_size'] * 100.0 / results['lpmem_size_limit']
|
| 82 |
|
| 83 |
+
output_text += f'<br>vmem_size = {vmem_size:0.3f} kB ({vmem_perc:0.1f}% of {vmem_size_limit:0.3f} kB limit)'
|
| 84 |
+
output_text += f'<br>lpmem_size = {lpmem_size:0.3f} kB ({lpmem_perc:0.1f}% of {lpmem_size_limit:0.3f} kB limit)'
|
| 85 |
|
| 86 |
|
| 87 |
return output_text
|
|
|
|
| 91 |
|
| 92 |
with gr.Blocks() as demo:
|
| 93 |
#gr.LoginButton()
|
| 94 |
+
text1 = gr.Markdown("SR100 Model Optimizer Demo - Compile a tflite model to SR100 and see if it fits")
|
| 95 |
#user_text = gr.Markdown("")
|
| 96 |
|
| 97 |
# Setup model inputs
|