Update README.md

README.md

@@ -52,7 +52,8 @@ for opt_state in OPT:
     subprocess.run(compile_command, shell=True, check=True)
     
     input_asm = ''
-    asm = read_file(output_file+'.s')
+    with open(output_file+'.s') as f:#original file
+        asm= f.read()
     asm = asm.split('Disassembly of section .text:')[-1].strip()
     for tmp in asm.split('\n'):
         tmp_asm = tmp.split('\t')[-1]#remove the binary code
@@ -79,7 +80,7 @@ with open(fileName +'_' + opt_state +'.asm','r') as f:#original file
     asm_func = f.read()
 inputs = tokenizer(asm_func, return_tensors="pt").to(model.device)
 with torch.no_grad():
-    outputs = model.generate(**inputs, max_new_tokens=200)
+    outputs = model.generate(**inputs, max_new_tokens=500)
 c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])
 ```