avoid gr.load

app.py

@@ -4,8 +4,30 @@ import os
 import re
 import subprocess
 import tempfile
+from transformers import pipeline
 
-model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
+
+MODEL_ID = "ejschwartz/oo-method-test-model-bylibrary"
+
+
+classifier = pipeline(
+    "text-classification",
+    model=MODEL_ID,
+)
+
+def run_model(text):
+    results = classifier(text, top_k=None, truncation=True)
+    if isinstance(results, dict):
+        results = [results]
+    if results and isinstance(results[0], list):
+        results = results[0]
+
+    confidences = [
+        {"label": entry["label"], "confidence": entry["score"]}
+        for entry in results
+    ]
+    best_label = max(confidences, key=lambda entry: entry["confidence"])["label"] if confidences else "unknown"
+    return {"label": best_label, "confidences": confidences}
 
 print("Model loaded")
 
@@ -121,7 +143,7 @@ with gr.Blocks() as demo:
         disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
         print("Evaluating model")
         try:
-            load_results = model.fn(disassembly_str)
+            load_results = run_model(disassembly_str)
         except Exception as e:
             print("Error evaluating model:", e)
             raise