Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
-import spaces
+#import spaces
 #import training
-import model_compiler
+#import model_compiler
+import sr100_model_compiler
 from huggingface_hub import HfApi
 from huggingface_hub import whoami
 
-compiler = model_compiler.model_compiler()
+#compiler = model_compiler.model_compiler()
 
 # Get top-level authorizations
 oauth_info = {'username' : None, 'token' : None}
@@ -31,8 +32,20 @@ def get_oauth_info(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken |
 
 def compile_model(model_name):
 
-    compiler.compile_model()
-
+    # Run the comparison
+    mode = 'hello_world.tflite'
+    out_dir = './tmp'
+    
+    
+    results = sr100_model_compiler.sr100_model_compiler(
+        model_file=model,
+        output_dir=f"{out_dir}",
+        model_loc=f"{model_loc}",
+    )
+    print(results)
+
+    default_config = sr100_model_compiler.sr100_default_config()
+    success, perf_data = sr100_model_compiler.sr100_check_model(results=results, config=default_config)
     
 #    try:
 #        epochs = int(epochs_text)
@@ -45,7 +58,7 @@ def compile_model(model_name):
     # Run the training
 #    return model.run_training(epochs=epochs, batch=batch)
 
-    return f"Compiled model {model_name}"
+    return f"Compiled model {success} {perf_data}"
 
 with gr.Blocks() as demo:
     gr.LoginButton()