Update app.py

app.py

@@ -3,29 +3,28 @@ import gradio as gr
 import tempfile
 import os
 import sr100_model_compiler
-from huggingface_hub import HfApi
-from huggingface_hub import whoami
-
+#from huggingface_hub import HfApi
+#from huggingface_hub import whoami
 # Get top-level authorizations
-oauth_info = {'username' : None, 'token' : None}
-api = HfApi()
-
-def get_oauth_info(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
-    global oauth_info
-    global api
-
-    print(f'profile = {profile}')
-    if profile is None:
-        oauth_info['username'] = None
-        oauth_info['token'] = None        
-        return "Please login to the Huggingface with login button"
-    else:
-        #print(f'Testing {profile} for username')
-        oauth_info['username'] = profile.username
-        oauth_info['token'] = oauth_token.token
-        org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
-
-        return print(f'{profile.username}: {org_names}')
+#oauth_info = {'username' : None, 'token' : None}
+#api = HfApi()
+#
+#def get_oauth_info(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
+#    global oauth_info
+#    global api
+
+#    print(f'profile = {profile}')
+#    if profile is None:
+#        oauth_info['username'] = None
+#        oauth_info['token'] = None        
+#        return "Please login to the Huggingface with login button"
+#    else:
+#        #print(f'Testing {profile} for username')
+#        oauth_info['username'] = profile.username
+#        oauth_info['token'] = oauth_token.token
+#        org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
+#
+#        return print(f'{profile.username}: {org_names}')
 
 
 def compile_model(model_name, vmem_value, lpmem_value):
@@ -61,7 +60,7 @@ def compile_model(model_name, vmem_value, lpmem_value):
             output_text = 'FAILURE, model does not fit on SR100'
 
         weights_size = results['weights_size'] / 1000.0
-        arena_size = results['arena_cache_size'] / 100.0
+        arena_size = results['arena_cache_size'] / 1000.0
         clock = results['core_clock']  / 1.0e6
         
         output_text += f'<br>clock = {clock:0.1f} MHz'
@@ -70,14 +69,15 @@ def compile_model(model_name, vmem_value, lpmem_value):
         output_text += f'<br>model loc = {results["model_loc"]}'
         output_text += f'<br>System config = {results["system_config"]}'
 
-
         vmem_size = results['vmem_size'] / 1000.0
         lpmem_size = results['lpmem_size'] / 1000.0
+        vmem_size_limit = results['vmem_size_limit'] / 1000.0
+        lpmem_size_limit = results['lpmem_size_limit'] / 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.3f} kB ({vmem_perc:0.1f}%)'
-        output_text += f'<br>lpmem_size = {lpmem_size:0.3f} kB ({lpmem_perc:0.1f}%)'
+        output_text += f'<br>vmem_size = {vmem_size:0.3f} kB ({vmem_perc:0.1f}% of {vmem_size_limit:0.3f} kB)'
+        output_text += f'<br>lpmem_size = {lpmem_size:0.3f} kB ({lpmem_perc:0.1f}% of {lpmem_size_limit:0.3f} kB)'
 
     
     return output_text
@@ -86,7 +86,7 @@ def compile_model(model_name, vmem_value, lpmem_value):
 model_choices = glob.glob('models/*.tflite')
     
 with gr.Blocks() as demo:
-    gr.LoginButton()
+    #gr.LoginButton()
     text1 = gr.Markdown("SR100 Model Compiler - Compile a tflite model to SR100")
     user_text = gr.Markdown("")