Improve model card: Add pipeline tag, library name, and paper link

README.md

@@ -3,8 +3,12 @@ license: mit
 tags:
 - decompile
 - binary
+pipeline_tag: text-generation
+library_name: transformers
 ---
 
+This repository contains the `LLM4Binary/llm4decompile-6.7b-v2` model. This model is associated with the paper [Decompile-Bench: Million-Scale Binary-Source Function Pairs for Real-World Binary Decompilation](https://huggingface.co/papers/2505.12668).
+
 ### 1. Introduction of LLM4Decompile
 
 LLM4Decompile aims to decompile x86 assembly instructions into C. The newly released V2 series are trained with a larger dataset (2B tokens) and a maximum token length of 4,096, with remarkable performance (up to 100% improvement) compared to the previous model.
@@ -53,6 +57,7 @@ Note: **Replace** func0 with the function name you want to decompile.
 import os
 import subprocess
 from tqdm import tqdm,trange
+import tempfile
 
 OPT = ["O0", "O1", "O2", "O3"]
 timeout_duration = 10
@@ -92,7 +97,8 @@ with tempfile.TemporaryDirectory() as temp_dir:
             c_decompile = f.read()
         c_func = []
         flag = 0
-        for line in c_decompile.split('\n'):
+        for line in c_decompile.split('
+'):
             if "Function: func0" in line:#**Replace** func0 with the function name you want to decompile.
                 flag = 1
                 c_func.append(line)
@@ -108,10 +114,14 @@ with tempfile.TemporaryDirectory() as temp_dir:
             if 'func0' in c_func[idx_tmp]:
                 break
         c_func = c_func[idx_tmp:]
-        input_asm = '\n'.join(c_func).strip()
-
-        before = f"# This is the assembly code:\n"#prompt
-        after = "\n# What is the source code?\n"#prompt
+        input_asm = '
+'.join(c_func).strip()
+
+        before = f"# This is the assembly code:
+"#prompt
+        after = "
+# What is the source code?
+"#prompt
         input_asm_prompt = before+input_asm.strip()+after
         with open(fileName +'_' + opt +'.pseudo','w',encoding='utf-8') as f:
             f.write(input_asm_prompt)
@@ -162,8 +172,10 @@ c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])
 with open(fileName +'_' + OPT[0] +'.pseudo','r') as f:#original file
     func = f.read()
 
-print(f'pseudo function:\n{func}')# Note we only decompile one function, where the original file may contain multiple functions
-print(f'refined function:\n{c_func_decompile}')
+print(f'pseudo function:
+{func}')# Note we only decompile one function, where the original file may contain multiple functions
+print(f'refined function:
+{c_func_decompile}')
 
 ```
 
@@ -172,4 +184,4 @@ This code repository is licensed under the MIT License.
 
 ### 5. Contact
 
-If you have any questions, please raise an issue.
+If you have any questions, please raise an issue.