Call helper

README.md

@@ -41,3 +41,4 @@ It's unclear why a1, a2, and result are not listed.
 ## Todo
 
 * Add field decoding (probably needs Docker)
+* ReSym's parser fails for functions with a name

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_client import Client
 import frontmatter
 import os
 import spaces
@@ -20,17 +21,7 @@ vardecoder_model = AutoModelForCausalLM.from_pretrained(
 #     "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16, device_map="auto"
 # )
 
-example = r"""__int64 __fastcall sub_410D81(__int64 a1, __int64 a2, __int64 a3)
-{
-int v4; // [rsp+20h] [rbp-20h] BYREF
-__int64 v5; // [rsp+28h] [rbp-18h]
-
-if ( !a1 || !a2 || !a3 )
-return 0LL;
-v4 = 5;
-v5 = a3;
-return sub_411142(a1, a2, &v4);
-}"""
+gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")
 
 examples = [
     ex.encode().decode("unicode_escape") for ex in open("examples.txt", "r").readlines()
@@ -80,6 +71,13 @@ def infer(code):
         skip_special_tokens=True,
         clean_up_tokenization_spaces=True,
     )
+
+    field_helper_result = gradio_client.predict(
+        decompiled_code=code,
+        api_name="/predict",
+    )
+    print(field_helper_result)
+
     # field_output = fielddecoder_model.generate(
     #     input_ids=input_ids,
     #     max_new_tokens=1024,

requirements.txt

@@ -1,3 +1,4 @@
 transformers
 accelerate
-python-frontmatter
+python-frontmatter
+gradio_client