cleanup

app.py

@@ -5,15 +5,10 @@ import re
 import subprocess
 import tempfile
 from transformers import pipeline
-from transformers.utils import logging as hf_logging
 
 
 MODEL_ID = "ejschwartz/oo-method-test-model-bylibrary"
 
-hf_logging.set_verbosity_debug()
-hf_logging.enable_default_handler()
-hf_logging.enable_explicit_format()
-
 classifier = pipeline(
     "text-classification",
     model=MODEL_ID,
@@ -33,8 +28,6 @@ def run_model(text):
     best_label = max(confidences, key=lambda entry: entry["confidence"])["label"] if confidences else "unknown"
     return {"label": best_label, "confidences": confidences}
 
-print("Model loaded")
-
 def get_all_dis(bname, addrs=None):
 
     anafile = tempfile.NamedTemporaryFile(prefix=os.path.basename(bname) + "_", suffix=".bat_ana")
@@ -121,8 +114,6 @@ with gr.Blocks() as demo:
 
     def file_change_fn(file, progress=gr.Progress()):
 
-        print("File change")
-
         if file is None:
             return {col: gr.update(visible=False),
                     all_dis_state: None}
@@ -142,16 +133,9 @@ with gr.Blocks() as demo:
         
     def function_change_fn(selected_fun, fun_data):
 
-        print("Selected function:", selected_fun)
-
         disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
-        print("Evaluating model")
-        try:
-            load_results = run_model(disassembly_str)
-        except Exception as e:
-            print("Error evaluating model:", e)
-            raise
-        print("Done with model")
+
+        load_results = run_model(disassembly_str)
         top_k = {e['label']: e['confidence'] for e in load_results['confidences']}
 
         return {disassembly: gr.update(value=disassembly_str),