Use .fn instead of __call__

app.py

@@ -10,7 +10,7 @@ import transformers
 import numpy
 print(numpy.__version__)
 
-pipe = transformers.pipeline(model="ejschwartz/oo-method-test-model-bylibrary", top_k=2)
+#pipe = transformers.pipeline(model="ejschwartz/oo-method-test-model-bylibrary", top_k=2)
 model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
 
 # gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models", examples=[
@@ -104,13 +104,13 @@ with gr.Blocks() as demo:
     def function_change_fn(selected_fun, fun_data):
 
         disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
-        model_results = pipe(disassembly_str)
-        load_results = model(disassembly_str)
+        #model_results = pipe(disassembly_str)
+        load_results = model.fn(disassembly_str)
         #top_k = {i["label"].split(", ")[0]: i["score"] for i in model_results}
-        disassembly_str = str(model_results) + str(load_results)
+        #disassembly_str = str(model_results) + str(load_results)
 
         return {disassembly: gr.Textbox.update(value=disassembly_str),
-                #clazz: gr.Label.update(top_k)
+                clazz: gr.Label.update(top_k)
                 #clazz: gr.Label.update(value="Method" if model_results[0]["label"] == "method" else "Function")}
         }