Add header

app.py

@@ -32,6 +32,13 @@ examples = json.load(open("humaneval_decompile_nova_6.7b.json", "r"))
 def predict(type, input_asm, _c_source):
 
     if "<func0>:" not in input_asm:
+
+        # Add a bogus function header if needed.
+        first_line = input_asm.split("\n")[0]
+        if "<" not in first_line or ">" not in first_line:
+            print("Adding bogus function header")
+            input_asm = "<func0>:\n" + input_asm
+
         normalized_asm = normalize(input_asm)
     else:
         normalized_asm = input_asm
@@ -78,13 +85,7 @@ def predict(type, input_asm, _c_source):
 
     return output
 
-example = """
-what.o:     file format elf64-x86-64
-
-
-Disassembly of section .text:
-
-0000000000000000 <mystery>:
+example = """0000000000000000 <mystery>:
    0:	f3 0f 1e fa          	endbr64 
    4:	55                   	push   %rbp
    5:	48 89 e5             	mov    %rsp,%rbp